mm

paddle. mm ( input, mat2, out=None, name=None ) [源代码]

用于两个输入矩阵的相乘。

两个输入的形状可为任意维度,但当任一输入维度大于 3 时,两个输入的维度必须相等。

如果原始 Tensor input 或 mat2 的秩为 1 且未转置,则矩阵相乘后的前置或附加维度 1 将移除。

参数

  • input (Tensor):输入变量,类型为 Tensor 或 LoDTensor。

  • mat2 (Tensor):输入变量,类型为 Tensor 或 LoDTensor。

  • out (Tensor,可选) – 指定存储运算结果的 Tensor。如果设置为 None 或者不设置,将创建新的 Tensor 存储运算结果,默认值为 None。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

  • Tensor,矩阵相乘后的结果。

*  1:

input: [B, ..., M, K], mat2: [B, ..., K, N]
out: [B, ..., M, N]

*  2:

input: [B, M, K], mat2: [B, K, N]
out: [B, M, N]

*  3:

input: [B, M, K], mat2: [K, N]
out: [B, M, N]

*  4:

input: [M, K], mat2: [K, N]
out: [M, N]

*  5:

input: [B, M, K], mat2: [K]
out: [B, M]

*  6:

input: [K], mat2: [K]
out: [1]

代码示例

import paddle
input = paddle.arange(1, 7).reshape((3, 2)).astype('float32')
mat2 = paddle.arange(1, 9).reshape((2, 4)).astype('float32')
out = paddle.mm(input, mat2)
print(out)
#        [[11., 14., 17., 20.],
#         [23., 30., 37., 44.],
#         [35., 46., 57., 68.]])