mm

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

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

    参数:

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

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

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

    代码示例

    1. import paddle
    2. mat2 = paddle.arange(1, 9).reshape((2, 4)).astype('float32')
    3. # [[11., 14., 17., 20.],
    4. # [23., 30., 37., 44.],