nn_init_eye_ | 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)
}
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.