1 torch.mm

只能进行2个2维矩阵的乘， 维度不对会报错。

2 torch.bmm

只能进行2个3维矩阵的乘，第1维是batch， 在后两个维度上进行矩阵乘。

3 torch.matmul

可以进行高维矩阵的乘， 在最后两个维度上进行矩阵乘。