cupy.eye#

cupy.eye(N, M=None, k=0, dtype=<class 'float'>, order='C')[source]#

返回一个二维数组，其对角线上是1，其余位置是0。

参数:

N (int) – 行数。
M (int) – 列数。默认为 M == N。
k (int) – 对角线索引。零表示主对角线，正索引表示上对角线，负索引表示下对角线。
dtype (数据类型, 可选) – 数据类型说明符。
order ({'C', 'F'}) – 行主序 (C 风格) 或列主序 (Fortran 风格)。

返回:

一个二维数组，给定对角线上填充1，其余位置填充0。

返回类型:

另请参阅