详解torch.Tensor的4种乘法_tensor矩阵乘法资源-CSDN文库

5星 · 超过95%的资源 84 浏览量 2020-12-17 02:15:28 上传评论收藏 52KB PDF 举报

资源推荐

资源详情

资源评论

详解详解torch.Tensor的的4种乘法种乘法

torch.Tensor有4种常见的乘法：*, torch.mul, torch.mm, torch.matmul. 本文抛砖引玉，简单叙述一下这4种乘法的区别，具体

使用还是要参照官方文档。

点乘点乘

a与b做*乘法，原则是如果a与b的size不同，则以某种方式将a或b进行复制，使得复制后的a和b的size相同，然后再将a和b

做element-wise的乘法的乘法。

下面以*标量和*一维向量为例展示上述过程。

* 标量标量

Tensor与标量k做*乘法的结果是Tensor的每个元素乘以k（相当于把k复制成与lhs大小相同，元素全为k的Tensor）.

>>> a = torch.ones(3,4)

>>> a

tensor([[1., 1., 1., 1.],

[1., 1., 1., 1.],

[1., 1., 1., 1.]])

>>> a * 2

tensor([[2., 2., 2., 2.],

[2., 2., 2., 2.],

[2., 2., 2., 2.]])

* 一维向量一维向量

Tensor与行向量做*乘法的结果是每列乘以行向量对应列的值（相当于把行向量的行复制，成为与lhs维度相同的Tensor）. 注

意此时要求Tensor的列数与行向量的列数相等。

>>> a = torch.ones(3,4)

>>> a

tensor([[1., 1., 1., 1.],

[1., 1., 1., 1.],

[1., 1., 1., 1.]])

>>> b = torch.Tensor([1,2,3,4])

>>> b

tensor([1., 2., 3., 4.])

>>> a * b

tensor([[1., 2., 3., 4.],

[1., 2., 3., 4.],

[1., 2., 3., 4.]])

Tensor与列向量做*乘法的结果是每行乘以列向量对应行的值（相当于把列向量的列复制，成为与lhs维度相同的Tensor）. 注

意此时要求Tensor的行数与列向量的行数相等。

>>> a = torch.ones(3,4)

>>> a

tensor([[1., 1., 1., 1.],

[1., 1., 1., 1.],

[1., 1., 1., 1.]])

>>> b = torch.Tensor([1,2,3]).reshape((3,1))

>>> b

tensor([[1.],

[2.],

[3.]])

>>> a * b

tensor([[1., 1., 1., 1.],

[2., 2., 2., 2.],

[3., 3., 3., 3.]])

* 矩阵矩阵

经Arsmart在评论区提醒，增补一个矩阵 * 矩阵的例子，感谢Arsmart的热心评论！

如果两个二维矩阵A与B做点积A * B，则要求A与B的维度完全相同，即A的行数=B的行数，A的列数=B的列数

>>> a = torch.tensor([[1, 2], [2, 3]])

>>> a * a

tensor([[1, 4],

[4, 9]])

本内容试读结束，登录后可阅读更多

下载后可阅读完整内容，剩余3页未读，立即下载

内容反馈

好运爆棚

2023-06-10

我觉得作者讲解得很清晰，但对于新手来说，可能需要花一点时间去理解。
IYA1738

2023-06-10

这篇文章详尽介绍了Torch.Tensor的4种乘法，让我对这个工具有了全新的认识。
生活教会我们

2023-06-10

这篇文章既简单易懂，又充满了实用的技巧，让人受益匪浅。
人亲卓玛

2023-06-10

文章虽然比较短，但是讲的很实用，解释得非常清楚，值得一读。
士多霹雳酱

2023-06-10

这篇文章提供了很多实用的案例和建议，帮助我更好地掌握了Tensor乘法的技巧。

前往

页

weixin_38529293

粉丝: 3
资源: 870

最新资源

资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！点击此处反馈

feedback-tip