Skip to main content

xavier_init

ruvector_gnn::tensor

Function xavier_init 

Source 
pub fn xavier_init(fan_in: usize, fan_out: usize) -> Vec<f32>
Expand description

Xavier/Glorot initialization for neural network weights

Samples from uniform distribution U(-a, a) where a = sqrt(6 / (fan_in + fan_out))

§Arguments

  • fan_in - Number of input units
  • fan_out - Number of output units

§Returns

A vector of initialized weights

§Panics

Panics if fan_in or fan_out is 0