点乘 torch.mul(a,b)二维矩阵乘 torch.mm(a,b)三维矩阵乘 torch.bmm(a,b)高维矩阵乘 torch.matmul(a,b)
https://blog.csdn.net/sunjinshengli/article/details/121901225