【Pytorch】进阶学习：实现矩阵乘法---torch.mul() 、 torch.mm() 和torch.matmul()的区别

【Pytorch】进阶学习：实现矩阵乘法—torch.mul() 、 torch.mm() 和torch.matmul()的区别

🌵文章目录🌵

• 🚀一、引言
• 🔍二、torch.mul()：元素级别的乘法
• 🔩三、torch.mm()：矩阵乘法（只适用于二维张量）
• 🎛️四、torch.matmul()：广义的矩阵乘法（适用于任意维度张量）
• 🔎五、总结与注意事项
• 🤝六、期待与你共同进步

🚀一、引言

在深度学习和神经网络的世界里，矩阵乘法是一项至关重要的操作。PyTorch作为目前最流行的深度学习框架之一，提供了多种矩阵乘法的实现方式。其中，torch.mul()、torch.mm()和torch.matmul()是三个常用的函数，但它们在用法和功能上却有所不同。本文将详细解释这三个函数的区别，并通过实例演示它们的使用方法。

🔍二、torch.mul()：元素级别的乘法

torch.mul()函数用于执行元素级别的乘法，即对应位置的元素相乘。这个函数对于两个形状相同的张量特别有用。

import torch

# 创建两个形状相同的张量
tensor1 = torch.tensor([[1, 2], [3, 4]])
tensor2 = torch.tensor([[5, 6], [7, 8]])

# 使用torch.mul()进行元素级别的乘法
result_mul = torch.mul(tensor1, tensor2)
print(result_mul)

输出：

tensor([[ 5, 12],
        [21, 32]])

如你所见，torch.mul()将tensor1和tensor2对应位置的元素相乘，得到一个新的张量。

🔩三、torch.mm()：矩阵乘法（只适用于二维张量）

torch.mm()函数用于执行矩阵乘法，但它只适用于二维张量（即矩阵）。如果你试图对高于二维的张量使用torch.mm()，将会得到一个错误。

# 创建两个二维张量
matrix1 = torch.tensor([[1, 2], [3, 4]])
matrix2 = torch.tensor([[5, 6], [7, 8]])

# 使用torch.mm()进行矩阵乘法
result_mm = torch.mm(matrix1, matrix2)
print(result_mm)

输出：

tensor([[19, 22],
        [43, 50]])

注意，矩阵乘法的规则是第一个矩阵的列数必须与第二个矩阵的行数相同。在上面的例子中，matrix1是一个2x2的矩阵，matrix2也是一个2x2的矩阵，所以它们可以进行矩阵乘法。

🎛️四、torch.matmul()：广义的矩阵乘法（适用于任意维度张量）

torch.matmul()函数提供了更广泛的矩阵乘法功能，它可以处理任意维度的张量。这个函数会按照张量的维度自动进行合适的乘法操作。

import torch

# 创建两个二维张量
matrix1 = torch.tensor([[1, 2], [3, 4]])
matrix2 = torch.tensor([[5, 6], [7, 8]])

# 使用torch.mm()进行矩阵乘法
result_mm = torch.mm(matrix1, matrix2)
print(result_mm)

# 对于二维张量，torch.matmul()与torch.mm()行为相同
result_matmul_2d = torch.matmul(matrix1, matrix2)
print(result_matmul_2d)

# 对于高于二维的张量，torch.matmul()可以执行广播和批量矩阵乘法
tensor3d_1 = torch.randn(3, 2, 4)  # 3个2x4的矩阵
tensor3d_2 = torch.randn(3, 4, 5)  # 3个4x5的矩阵

# 批量矩阵乘法
result_matmul_3d = torch.matmul(tensor3d_1, tensor3d_2)
print(result_matmul_3d.shape)  # 输出应为(3, 2, 5)，表示3个2x5的矩阵

输出：

tensor([[19, 22],
        [43, 50]])
tensor([[19, 22],
        [43, 50]])
torch.Size([3, 2, 5])

torch.matmul()函数非常灵活，它可以处理各种复杂的张量乘法场景。

🔎五、总结与注意事项

总结一下，torch.mul()、torch.mm()和torch.matmul()这三个函数的主要区别在于它们处理张量的方式和维度要求不同。torch.mul()执行的是元素级别的乘法，要求输入张量形状相同；torch.mm()执行的是标准的矩阵乘法，只适用于二维张量；而torch.matmul()则提供了更广义的矩阵乘法，可以处理任意维度的张量，包括批量矩阵乘法。

在使用这些函数时，需要注意以下几点：

1. 确保输入张量的形状符合函数的要求，否则可能会引发错误。
2. 对于矩阵乘法，需要注意矩阵的维度匹配问题，即第一个矩阵的列数必须等于第二个矩阵的行数。
3. 在进行批量矩阵乘法时，使用torch.matmul()可以方便地处理多个矩阵的乘法运算。

🤝六、期待与你共同进步

🌱 亲爱的读者，非常感谢你每一次的停留和阅读！你的支持是我们前行的最大动力！🙏

🌐 在这茫茫网海中，有你的关注，我们深感荣幸。你的每一次点赞👍、收藏🌟、评论💬和关注💖，都像是明灯一样照亮我们前行的道路，给予我们无比的鼓舞和力量。🌟

📚 我们会继续努力，为你呈现更多精彩和有深度的内容。同时，我们非常欢迎你在评论区留下你的宝贵意见和建议，让我们共同进步，共同成长！💬

💪 无论你在编程的道路上遇到什么困难，都希望你能坚持下去，因为每一次的挫折都是通往成功的必经之路。我们期待与你一起书写编程的精彩篇章！ 🎉

🌈 最后，再次感谢你的厚爱与支持！愿你在编程的道路上越走越远，收获满满的成就和喜悦！祝你编程愉快！🎉

关于Python学习指南

---

如果想要系统学习Python、Python问题咨询，或者考虑做一些工作以外的副业，都可以扫描二维码添加微信，围观朋友圈一起交流学习。

我们还为大家准备了Python资料和副业项目合集，感兴趣的小伙伴快来找我领取一起交流学习哦！

学好 Python 不论是就业还是做副业赚钱都不错，但要学会 Python 还是要有一个学习规划。最后给大家分享一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！

包括：Python激活码+安装包、Python web开发，Python爬虫，Python数据分析，人工智能、自动化办公等学习教程。带你从零基础系统性的学好Python！

👉Python所有方向的学习路线👈

Python所有方向路线就是把Python常用的技术点做整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。（全套教程文末领取）

👉Python学习视频600合集👈

观看零基础学习视频，看视频学习是最快捷也是最有效果的方式，跟着视频中老师的思路，从基础到深入，还是很容易入门的。

温馨提示：篇幅有限，已打包文件夹，获取方式在：文末

👉Python70个实战练手案例&源码👈

光学理论是没用的，要学会跟着一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

👉Python大厂面试资料👈

我们学习Python必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有阿里大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

👉Python副业兼职路线&方法👈

学好 Python 不论是就业还是做副业赚钱都不错，但要学会兼职接单还是要有一个学习规划。

👉 这份完整版的Python全套学习资料已经上传，朋友们如果需要可以扫描下方CSDN官方认证二维码或者点击链接免费领取【保证100%免费】