mm
用于两个输入矩阵的相乘。
两个输入的形状可为任意维度,但当任一输入维度大于3时,两个输入的维度必须相等。
参数:
input (Tensor) : 输入变量,类型为 Tensor 或 LoDTensor。
out (Tensor, 可选) – 指定存储运算结果的Tensor。如果设置为None或者不设置,将创建新的Tensor存储运算结果,默认值为None。
name (str,可选)- 具体用法请参见 ,一般无需设置,默认值为None。
代码示例:
import paddle
mat2 = paddle.arange(1, 9).reshape((2, 4)).astype('float32')
# [[11., 14., 17., 20.],
# [23., 30., 37., 44.],