mindspore.mint.mm

mindspore.mint.mm(input, mat2)

计算两个矩阵的乘积。

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

说明

此函数不能支持广播。若需要可广播的方法，请参考 mindspore.mint.matmul()。 input 的最后一维度必须和 mat2 的第一维度相等。

警告

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

参数：

• input (Tensor) - 第一个矩阵tensor。

• mat2 (Tensor) - 第二个矩阵tensor。

返回：

Tensor

支持平台：

Ascend

样例：

>>> import mindspore
>>> x1 = mindspore.mint.rand(2, 3)
>>> x2 = mindspore.mint.rand(3, 4)
>>> mindspore.mint.mm(x1, x2).shape
(2, 4)