mindspore.ops.eye ================== .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/ops/mindspore.ops.func_eye.rst :alt: 查看源文件 .. py:function:: mindspore.ops.eye(n, m=None, dtype=None) 创建一个主对角线上元素为1,其余元素为0的Tensor。 .. note:: Ascend平台支持返回Tensor的数据类型包括float16, float32, int8, int16, int32, int64, uint8和bool。 参数: - **n** (int) - 指定返回Tensor的行数。仅支持常量值。 - **m** (int,可选) - 指定返回Tensor的列数。仅支持常量值。默认值为None,返回Tensor的列数默认与n相等。 - **dtype** (mindspore.dtype,可选) - 指定返回Tensor的数据类型。数据类型必须是 `bool_ `_ 或 `number `_ 。默认值为 ``None`` ,返回Tensor的数据类型默认为mindspore.float32。 返回: Tensor,主对角线上为1,其余的元素为0。它的shape由 `n` 和 `m` 指定。数据类型由 `dtype` 指定。 异常: - **TypeError** - `m` 或 `n` 不是int。 - **ValueError** - `m` 或 `n` 小于0。