mindspore.numpy.identity
- mindspore.numpy.identity(n, dtype=mstype.float32)[源代码]
返回单位数组。主对角线上全为1。
- 参数:
n (int) - 设置的输出数组的行数和列数,必须大于0。
dtype (Union[mindspore.dtype, str], 可选) - 指定的Tensor
dtype
,默认值:mstype.float32
。
- 返回:
Tensor,shape为
(n, n)
,除主对角线上值为1,其余所有元素都等于0。- 异常:
TypeError - 如果输入参数非上述给定的类型。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore.numpy as np >>> print(np.identity(2)) [[1. 0.] [0. 1.]]