xorbits.numpy.eye#

xorbits.numpy.eye(N, M=None, k=0, dtype=None, sparse=False, gpu=None, chunk_size=None, order='C')[源代码]#

Return a 2-D array with ones on the diagonal and zeros elsewhere.

参数

  • 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 a negative value to a lower diagonal.

  • dtype (data-type, optional) – Data-type of the returned array.

  • order ({'C', 'F'}, optional) –

    Whether the output should be stored in row-major (C-style) or column-major (Fortran-style) order in memory.

    1.14.0(numpy) 新版功能.

  • like (array_like, optional (Not supported yet)) –

    Reference object to allow the creation of arrays which are not NumPy arrays. If an array-like passed in as like supports the __array_function__ protocol, the result will be defined by it. In this case, it ensures the creation of an array object compatible with that passed in via this argument.

    1.20.0(numpy) 新版功能.

返回

I – An array where all elements are equal to zero, except for the k-th diagonal, whose values are equal to one.

返回类型

ndarray of shape (N,M)

参见

identity

(almost) equivalent function

diag

diagonal 2-D array from a 1-D array specified by the user.

实际案例

>>> np.eye(2, dtype=int)  
array([[1, 0],
       [0, 1]])
>>> np.eye(3, k=1)  
array([[0.,  1.,  0.],
       [0.,  0.,  1.],
       [0.,  0.,  0.]])

This docstring was copied from numpy.