Tensor.eye()

Returns the identity tensor of size n x n.

Usage

from tinygrad.tensor import Tensor
 
a1 = Tensor.eye(5)
 
print(a1.numpy())

[[1. 0. 0. 0. 0.]
 [0. 1. 0. 0. 0.]
 [0. 0. 1. 0. 0.]
 [0. 0. 0. 1. 0.]
 [0. 0. 0. 0. 1.]]

;