mindspore.ops.eye

mindspore.ops.eye(n, m, t)[源代码]

创建一个主对角线上元素为1，其余元素为0的Tensor。

说明

结合ReverseV2算子可以得到一个反对角线为1的Tensor，但是目前ReverseV2算子只支持Ascend和GPU平台。

参数：

  • n (int) - 指定返回Tensor的行数。仅支持常量值。

  • m (int) - 指定返回Tensor的列数。仅支持常量值。

  • t (mindspore.dtype) - 指定返回Tensor的数据类型。数据类型必须是 bool_ 或 number 。

返回：

Tensor，主对角线上为1，其余的元素为0。它的shape由 n 和 m 指定。数据类型由 t 指定。

异常：

  • TypeError - m 或 n 不是int。

  • ValueError - m 或 n 小于1。

支持平台：

Ascend GPU CPU

样例：

>>> output = ops.eye(2, 2, mindspore.int32)
>>> print(output)
[[1 0]
 [0 1]]
>>> print(output.dtype)
Int32
>>> output = ops.eye(1, 2, mindspore.float64)
>>> print(output)
[[1. 0.]]
>>> print(output.dtype)
Float64