torch.mul和直接用运算符 * 都是点乘(可广播)
torch.mm只能二维矩阵相乘,而torch.matmul可以广播到多维
参见:
https://blog.youkuaiyun.com/weixin_42105432/article/details/100691592
pytorch点乘&高维矩阵乘法
最新推荐文章于 2025-06-13 19:41:35 发布
本文详细解析了PyTorch中矩阵运算的三种方式:torch.mul与运算符*实现的点乘,torch.mm用于二维矩阵乘法,以及更通用的torch.matmul支持多维矩阵的广播乘法。对比不同函数的适用场景,帮助读者深入理解PyTorch的矩阵运算机制。
4429

被折叠的 条评论
为什么被折叠?



