numpy.eye()
numpy.eye
-
numpy.eye(N, M=None, k=0, dtype=[source]) -
Return a 2-D array with ones on the diagonal and zeros elsewhere.
Parameters: N : int
Number of rows in the output.
M : int, optional
Number of columns in the output. If None, defaults to
N.k : int, optional
Index of the diagonal: 0 (the default) refers to the main diagonal, a positive value refers to an upper diagonal, and