Enum burn::tensor::DistributionSamplerKind

pub enum DistributionSamplerKind<E>
where
    Standard: Distribution<E>,
    E: SampleUniform,
{
    Standard(Standard),
    Uniform(Uniform<E>),
    Bernoulli(Bernoulli),
    Normal(Normal<f64>),
}

Distribution sampler kind for random value of a tensor.

Variants

Standard(Standard)

Standard distribution.

Uniform(Uniform<E>)

Uniform distribution.

Bernoulli(Bernoulli)

Bernoulli distribution.

Normal(Normal<f64>)

Normal distribution.