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()
。Ascend上,暂不支持float64类型。
- 参数:
input (Tensor) - 矩阵相乘的第一个矩阵, input 的最后一维度必须和 mat2 的第一维度相等。
mat2 (Tensor) - 矩阵相乘的第二个矩阵, input 的最后一维度必须和 mat2 的第一维度相等。
- 返回:
Tensor,输入的矩阵乘积。
- 异常:
ValueError - input 的最后一维度和 mat2 的倒数第二维度不相等。
ValueError - input 或者 mat2 不是一个Tensor。
- 支持平台:
Ascend
GPU
CPU
样例:
>>> import mindspore as ms >>> from mindspore import ops >>> 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 = ops.mm(x1, x2) >>> print(out.shape) (2, 4)