cupy.identity#

cupy.identity(n, dtype=<class 'float'>)[source]#

返回一个二维单位矩阵(恒等数组)。

等价于 eye(n, n, dtype)

参数:
  • n (int) – 行数和列数。

  • dtype (数据类型, 可选) – 数据类型指定符。

返回:

一个二维单位矩阵。

返回类型:

cupy.ndarray

另请参阅

numpy.identity()