Enum InitMode 

pub enum InitMode {
    Zeros,
    Ones,
    Constant(f32),
    Uniform,
    UniformRange(f32, f32),
    Normal(f32, f32),
    XavierUniform,
    XavierNormal,
    KaimingUniform,
    KaimingNormal,
    Orthogonal(f32),
}

Initialization strategies as an enum for dynamic selection.

Variants

Zeros

Zeros initialization.

Ones

Ones initialization.

Constant(f32)

Constant value initialization.

Uniform

Uniform random initialization.

UniformRange(f32, f32)

Uniform random in range.

Normal(f32, f32)

Normal distribution.

XavierUniform

Xavier/Glorot uniform.

XavierNormal

Xavier/Glorot normal.

KaimingUniform

Kaiming/He uniform.

KaimingNormal

Kaiming/He normal.

Orthogonal(f32)

Orthogonal.

Implementations

impl InitMode

pub fn init(&self, fan_out: usize, fan_in: usize) -> Tensor<f32>

Initializes a tensor using this mode.

Trait Implementations

impl Clone for InitMode

fn clone(&self) -> InitMode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for InitMode

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

Formats the value using the given formatter.

impl PartialEq for InitMode

fn eq(&self, other: &InitMode) -> 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 Copy for InitMode

impl StructuralPartialEq for InitMode