torch.matmul

 

0阶张量是一个标量。

1阶张量是一个向量。

2阶张量是一个矩阵。

如果俩个张量都是一维度，输出的结果是标量相乘。

如果俩个张量都是矩阵，输出的结果是矩阵乘积。

如果第一个是二维，第二个是一维张量，输出结果将第一个参数每一行看成一个向量与第二个进行向量积。（只有他需要注意）

如果第一个是一维张量，第二个是二维张量，输出结果也是矩阵乘积。