Skip to main content

xavier_uniform

Function xavier_uniform 

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

Xavier (Glorot) uniform initialization.

Values drawn from U(-limit, limit) where limit = gain * sqrt(6 / (fan_in + fan_out)).