pytorch的mm()函数

最新推荐文章于 2024-03-14 23:22:10 发布

Vertira

最新推荐文章于 2024-03-14 23:22:10 发布

阅读量1.9k

点赞数 1

分类专栏： pytorch 文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/Vertira/article/details/131460596

版权

pytorch 专栏收录该内容

86 篇文章 20 订阅

订阅专栏

torch.mul()执行的是元素级相乘，也称为Hadamard乘积，保持输入矩阵的维度不变。torch.mm()则进行矩阵乘法，要求特定的维度匹配，返回一个新的矩阵。torch.matmul()更通用，可处理多维矩阵。

摘要由CSDN通过智能技术生成

torch.mul(a, b)
torch.mul(a, b) 是矩阵 a和 b 对应位相乘，a 和 b 的维度必须相等，比如a 的维度是 (1,2)，b 的维度是(1,2)，返回的仍是(1,2)的矩阵，和 a ∗ b 效果相同。其实这个矩阵或者向量的乘法操作就是 Hadamard 乘积哈达玛乘积，具体操作细节查看如下。这个操作跟乘以b一样。

>>> a =torch.tensor([1,2,3,4])
>>> b = torch.tensor([5,6,7,8])
>>> c = torch.mul(a,b)
>>> c
tensor([ 5, 12, 21, 32])
>>> c1 = torch.mul(b,a)
>>> c1
tensor([ 5, 12, 21, 32])
>>>
# a*b  和b*a一样

矩阵的尺寸不变，