Function neuronika::nn::init::eye

pub fn eye(param: &Learnable<Ix2>)

Fills the matrix differentiable leaf variable with the identity matrix.

Preserves the identity of the inputs in Linear layers, where as many inputs are preserved as possible.

Arguments

param - differentiable variable to initialize.