/
envoy.istio
/
mindspore
Обзор
Документация
Войти
/
envoy.istio
/
mindspore
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
docs/api/api_python/ops/mindspore.ops.func_mm.rst
23 строки
989 B
zhanghanLeo
[bugfix]: modify ops.mm testcase for the Ascend GE doesn't support fp64.
26 мар 2024, 12:53
26 мар 2024, 12:53
0172552
Код
Авторство
О чём код?
mindspore.ops.mm ================= .. py:function:: mindspore.ops.mm(input, mat2) 计算两个矩阵的乘积。 如果 `input` 是一个 :math:`(n \times m)` 的Tensor， `mat2` 是一个 :math:`(m \times p)` 的Tensor， `out` 则会是一个 :math:`(n \times p)` 的Tensor。 .. note:: - 此函数不能支持广播。若需要可广播的方法，请参考 :func:`mindspore.ops.matmul`。 - Ascend上，暂不支持float64类型。 参数： - **input** (Tensor) - 矩阵相乘的第一个矩阵， `input` 的最后一维度必须和 `mat2` 的第一维度相等。 - **mat2** (Tensor) - 矩阵相乘的第二个矩阵， `input` 的最后一维度必须和 `mat2` 的第一维度相等。 返回： Tensor，输入的矩阵乘积。 异常： - **ValueError** - `input` 的最后一维度和 `mat2` 的倒数第二维度不相等。 - **ValueError** - `input` 或者 `mat2` 不是一个Tensor。