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