Enum burn_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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for Distribution

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for Distribution

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

Tests for self and other values to be equal, and is used by ==.

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

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

impl Serialize for Distribution

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for Distribution

impl StructuralPartialEq for Distribution