mindspore.numpy.eye

mindspore.numpy.eye(N, M=None, k=0, dtype=mstype.float32)

返回一个对角线上值为1，其他位置为0的二维Tensor。

参数：

• N (int) - 输出二维Tensor的行数。 输入值必须大于0。

• M (int, 可选) - 输出二维Tensor的列数。默认值： None ，如果输入为 None ，则默认值： N ，输入值必须大于0。

• k (int, 可选) - 对角线的索引。默认值： 0 ，即主对角线。正值表示上对角线，负值表示下对角线。

• dtype (Union[mindspore.dtype, str], 可选) - 指定的Tensor dtype 。默认值： mstype.float32 。

返回：

Tensor，shape为(N, M)。其中，除第 k 个对角线值等于1外，其余所有元素都等于0。

异常：

• TypeError - 如果输入参数非给定的数据类型。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> print(np.eye(2, 2))
[[1. 0.]
[0. 1.]]