pytorch矩阵乘法总结

点乘 torch.mul(a,b)
二维矩阵乘 torch.mm(a,b)
三维矩阵乘 torch.bmm(a,b)
高维矩阵乘 torch.matmul(a,b)

https://blog.csdn.net/sunjinshengli/article/details/121901225