Identity

class brainstate.nn.init.Identity(value=1.0, unit=Unit('1'))

Returns the identity matrix.

This initializer was proposed in (Le, et al., 2015) [1].

Parameters:

value (float) – The optional scaling factor.

Returns:

shape – The weight shape/size.

Return type:

tuple of int

References

[1]

Le, Quoc V., Navdeep Jaitly, and Geoffrey E. Hinton. “A simple way to initialize recurrent networks of rectified linear units.” arXiv preprint arXiv:1504.00941 (2015).