mindspore.numpy.eye ================================= .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/br_base/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/br_base/docs/api/api_python/numpy/mindspore.numpy.eye.rst :alt: 查看源文件 .. py:function:: 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** - 如果输入参数非给定的数据类型。