Enum burn_core::tensor::Distribution

pub enum Distribution {
    Default,
    Bernoulli(f64),
    Uniform(f64, f64),
    Normal(f64, f64),
}

Distribution for random value of a tensor.

Variants

Default

Uniform distribution from 0 (inclusive) to 1 (exclusive).

Bernoulli(f64)

Bernoulli distribution with the given probability.

Uniform(f64, f64)

Uniform distribution. The range is inclusive.

Normal(f64, f64)

Normal distribution with the given mean and standard deviation.

Implementations

impl Distribution

pub fn sampler<R, E>(self, rng: &mut R) -> DistributionSampler<'_, E, R>where R: RngCore, E: Element + SampleUniform, Standard: Distribution<E>,

Creates a new distribution sampler.

Arguments

• rng - The random number generator.

Returns

The distribution sampler.

Trait Implementations

impl Clone for Distribution

fn clone(&self) -> Distribution

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Distribution

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl PartialEq for Distribution

fn eq(&self, other: &Distribution) -> bool

This method tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

This method tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Copy for Distribution

impl StructuralPartialEq for Distribution