本文详细解析了PyTorch中矩阵运算的三种方式：torch.mul与运算符*实现的点乘，torch.mm用于二维矩阵乘法，以及更通用的torch.matmul支持多维矩阵的广播乘法。对比不同函数的适用场景，帮助读者深入理解PyTorch的矩阵运算机制。
torch.mul和直接用运算符 * 都是点乘（可广播）
torch.mm只能二维矩阵相乘，而torch.matmul可以广播到多维
参见：
https://blog.youkuaiyun.com/weixin_42105432/article/details/100691592
pytorch点乘&高维矩阵乘法
torch.mul和直接用运算符 * 都是点乘（可广播）
torch.mm只能二维矩阵相乘，而torch.matmul可以广播到多维
参见：
https://blog.youkuaiyun.com/weixin_42105432/article/details/100691592
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
