Only use for square 2D matrices.

initializer_identity(gain = 1)

Arguments

gain

Multiplicative factor to apply to the identity matrix