cupy.eye¶
-
cupy.
eye
(N, M=None, k=0, dtype=<class 'float'>, order='C')[source]¶ Returns a 2-D array with ones on the diagonals and zeros elsewhere.
- Parameters
N (int) – Number of rows.
M (int) – Number of columns.
M == N
by default.k (int) – Index of the diagonal. Zero indicates the main diagonal, a positive index an upper diagonal, and a negative index a lower diagonal.
dtype – Data type specifier.
order ({'C', 'F'}) – Row-major (C-style) or column-major (Fortran-style) order.
- Returns
A 2-D array with given diagonals filled with ones and zeros elsewhere.
- Return type
See also