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