Home / Class/ sampler Class — pytorch Architecture

sampler Class — pytorch Architecture

Architecture documentation for the sampler class in Distributions.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/Distributions.h lines 48–55

template<typename scalar_t, typename sampler_t>
struct BaseSampler {
  sampler_t sampler;
  C10_DEVICE BaseSampler(const sampler_t& sampler): sampler(sampler) {}
  C10_DEVICE scalar_t sample() {
    return sampler();
  }
};

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free