Enum tch::nn::Init

pub enum Init {
    Const(f64),
    Randn {
        mean: f64,
        stdev: f64,
    },
    Uniform {
        lo: f64,
        up: f64,
    },
    KaimingUniform,
}

Variable initializations.

Variants

Const(f64)

Constant value.

Randn

Random normal with some mean and standard deviation.

Fields of Randn

mean: f64stdev: f64

Uniform

Uniform initialization between some lower and upper bounds.

Fields of Uniform

lo: f64up: f64

KaimingUniform

Kaiming uniform initialization.

Methods

impl Init

pub fn set(self, tensor: &mut Tensor)

Re-initializes an existing tensor with the specified initialization

Trait Implementations

impl Clone for Init

fn clone(&self) -> Init

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Copy for Init

impl Debug for Init

fn fmt(&self, f: &mut Formatter) -> Result

Formats the value using the given formatter.