return 2D identical matrix with size N
np.eye(3) # [[1,0,0], [0,1,0], [0,0,1]]
numpy.eye — NumPy v2.1 Manual
Return a 2-D array with ones on the diagonal and zeros elsewhere.
https://numpy.org/doc/stable/reference/generated/numpy.eye.html
Seonglae Cho
Seonglae Chonp.eye(3) # [[1,0,0], [0,1,0], [0,0,1]]