tensorplay.nn.init.eye_

tensorplay.nn.init.eye_(tensor)[source]

Fill the 2-D tensor in-place with the identity matrix.

As many inputs as possible are preserved through a Linear layer.