Function he_init
Source pub fn he_init(fan_in: usize) -> Vec<f32>
Expand description
He initialization for ReLU networks
Samples from normal distribution N(0, sqrt(2 / fan_in))
§Arguments
fan_in - Number of input units
§Returns
A vector of initialized weights
§Panics
Panics if fan_in is 0