Skip to main content

kaiming_normal

Function kaiming_normal 

Source
pub fn kaiming_normal(
    shape: &[usize],
    gain: f64,
    mode: FanMode,
    rng: &mut InitRng,
) -> Result<ArrayD<f64>, InitError>
Expand description

Kaiming (He) normal initialization.

Values drawn from N(0, std) where std = gain / sqrt(fan).