nn_init_eye_ {torch} | R Documentation |
Fills the 2-dimensional input Tensor
with the identity matrix.
Preserves the identity of the inputs in Linear
layers, where as
many inputs are preserved as possible.
nn_init_eye_(tensor)
tensor |
a 2-dimensional torch tensor. |
if (torch_is_installed()) {
w <- torch_empty(3, 5)
nn_init_eye_(w)
}