pytorch点乘&高维矩阵乘法

torch.mul和直接用运算符 * 都是点乘(可广播)
torch.mm只能二维矩阵相乘,而torch.matmul可以广播到多维
参见:
https://blog.csdn.net/weixin_42105432/article/details/100691592

你可能感兴趣的:(pytorch代码积累)