Enum engram::nn::Initializer

pub enum Initializer {
    Xavier,
    Kaiming,
    Constant(f64),
}

Initialization methods for weight matrices.

Variants

Xavier

Xavier (Glorot) initialization. Gaussian, µ = 0, σ = √[2 / (f_in + f_out)]

Kaiming

Kaiming (He) initialization. Gaussian, µ = 0, σ = √[2 / f_in]

Constant(f64)

Constant initialization. All weights are set to the given value.

Implementations

impl Initializer

pub fn initialize(&self, f_in: usize, f_out: usize) -> Tensor2D

Initializes a 2D tensor with random values based on the specified initialization technique.

Examples

let xavier = Initializer::Xavier.initialize(2, 3);
let kaiming = Initializer::Kaiming.initialize(5, 3);

Trait Implementations

impl Debug for Initializer

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

Formats the value using the given formatter.

impl PartialEq for Initializer

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