Function neuronika::nn::init::calculate_fan_in_fan_out

pub fn calculate_fan_in_fan_out<D: Dimension>(
    param: &Learnable<D>
) -> (f32, f32)

Returns the fan_in and the fan_out.

For MLPs fan_in and fan_out are respectively the number of inputs and outputs to an hidden unit of the layer. For CNNs however, the number of input feature maps and the size of the receptive field must be taken into account .

Arguments

param - differentiable variable for which the fan in and the fan out must be calculated.