mindspore.ops.mm

mindspore.ops.mm(input, mat2)[源代码]

计算两个矩阵的乘积。

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

说明

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

参数:
  • input (Tensor) - 矩阵相乘的第一个矩阵, input 的最后一维度必须和 mat2 的第一维度相等。

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

返回:

Tensor,输入的矩阵乘积。

异常:
  • ValueError - input 的最后一维度和 mat2 的倒数第二维度不相等。

  • ValueError - input 或者 mat2 不是一个矩阵。

支持平台:

Ascend GPU CPU

样例:

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