Skip to main content

kaiming_normal

Function kaiming_normal 

Source
pub fn kaiming_normal(
    shape: Vec<usize>,
    fan_in: usize,
    seed: u64,
) -> Result<Tensor, ModelError>
Expand description

Kaiming (He) normal initialization.

Fills with values from N(0, std) where std = sqrt(2 / fan_in).