mindspore.Tensor.mm

mindspore.Tensor.mm(mat2)

计算两个矩阵的乘积。

如果 self 是一个 \((n \times m)\) 的Tensor， mat2 是一个 \((m \times p)\) 的Tensor， out 则会是一个 \((n \times p)\) 的Tensor。

说明

• 此函数不能支持广播。若需要可广播的方法，请参考 mindspore.ops.matmul()。

警告

这是一个实验性API，后续可能修改或删除。

参数：

• mat2 (Tensor) - 矩阵相乘的第二个矩阵， self 的最后一维度必须和 mat2 的第一维度相等。

返回：

Tensor，输入的矩阵乘积。

异常：

• TypeError - self 或者 mat2 不是一个Tensor。

• RuntimeError - self 的最后一维度和 mat2 的倒数第二维度不相等。

• RuntimeError - self 或者 mat2 不是float16、float32或bfloat16类型。

支持平台：

Ascend

样例：

>>> import mindspore as ms
>>> import numpy as np
>>> x1 = ms.Tensor(np.random.rand(2, 3), ms.float32)
>>> x2 = ms.Tensor(np.random.rand(3, 4), ms.float32)
>>> out = x1.mm(x2)
>>> print(out.shape)
(2, 4)