torch.mul和直接用运算符 * 都是点乘(可广播)
torch.mm只能二维矩阵相乘,而torch.matmul可以广播到多维
参见:
https://blog.youkuaiyun.com/weixin_42105432/article/details/100691592
pytorch点乘&高维矩阵乘法
最新推荐文章于 2025-05-03 16:56:44 发布
torch.mul和直接用运算符 * 都是点乘(可广播)
torch.mm只能二维矩阵相乘,而torch.matmul可以广播到多维
参见:
https://blog.youkuaiyun.com/weixin_42105432/article/details/100691592