Skip to main content

xavier_uniform

Function xavier_uniform 

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

Xavier (Glorot) uniform initialization.

Fills with values from U(-bound, bound) where bound = sqrt(6 / (fan_in + fan_out)).