/
envoy.istio
/
mindspore
Обзор
Документация
Войти
/
envoy.istio
/
mindspore
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
docs/api/api_python/ops/mindspore.ops.Eye.rst
18 строк
713 B
zlq2020
fix eye doc
08 мар 2024, 05:48
08 мар 2024, 05:48
8ec284c
Код
Авторство
О чём код?
mindspore.ops.Eye ================== .. py:class:: mindspore.ops.Eye 创建一个主对角线上元素为1，其余元素为0的Tensor。 .. note:: Ascend平台支持返回Tensor的数据类型包括float16, float32, int8, int16, int32, int64, uint8和bool。 更多参考详见 :func:`mindspore.ops.eye`。 输入： - **n** (int) - 指定返回Tensor的行数。仅支持常量值。 - **m** (int) - 指定返回Tensor的列数。仅支持常量值。 - **t** (mindspore.dtype) - 指定返回Tensor的数据类型。 输出： Tensor，主对角线上为1，其余的元素为0。它的shape由 `n` 和 `m` 指定。数据类型由 `t` 指定。