mindspore.ops.mm

mindspore.ops.mm(input, mat2)

计算两个输入的矩阵乘积。

若 input 的shape为 \((n \times m)\) ， mat2 的shape为 \((m \times p)\) ， out 的shape为 \((n \times p)\) 。

说明

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

• Ascend平台，不支持float64类型。

参数：

• input (Tensor) - 第一个输入tensor。

• mat2 (Tensor) - 第二个输入tensor。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> out = mindspore.ops.mm(mindspore.ops.ones((2, 3)), mindspore.ops.ones((3, 4)))
>>> print(out.shape)
(2, 4)