主要参考PyTorch入门笔记-张量相乘matmul函数02 - 腾讯云开发者社区-腾讯云 (tencent.com)
我自己认为容易记住的点:
1、维数不同时,扩张的是不会相乘的那个维度。例如:一个二维为(2,3)的张量和一个一维size为3的张量相乘,则扩张的是第二维度,即(2,3)x(3,.)=(2,.)再把扩张的此维度删去,即为一维张量,size为2。反过来,亦是如此,一个一维size为2为的张量和一个二维(2,3)张量相乘,则扩张的是第二维度,即(.,2)x(2,3)=(.,3)再把扩张的此维度删去,即为一维张量,size为3。注意不可以(.,3)x(2,3)。
2、