Struct burn_tensor::DistributionSampler

pub struct DistributionSampler<'a, E, R>where
    Standard: Distribution<E>,
    E: SampleUniform,
    R: RngCore,{ /* private fields */ }

Distribution sampler for random value of a tensor.

Implementations

impl<'a, E, R> DistributionSampler<'a, E, R>where Standard: Distribution<E>, E: SampleUniform, R: RngCore,

pub fn new(kind: DistributionSamplerKind<E>, rng: &'a mut R) -> Self

Constructs a new DistributionSampler.

impl<'a, E, R> DistributionSampler<'a, E, R>where Standard: Distribution<E>, E: SampleUniform + Element, R: RngCore,

pub fn sample(&mut self) -> E

Sames a random value from the distribution.