Pytorch的乘法是怎样的

Pytorch的乘法是怎样的，针对这个问题，这篇文章详细介绍了相对应的分析和解答，希望可以帮助更多想解决这个问题的小伙伴找到更简单易行的方法。

a = torch.tensor([[1, 2, 3], [1, 2, 3], [1, 2, 3]])
b = torch.tensor([[1, 1, 1], [2, 2, 2], [3, 3, 3]])
c = a * b
d = torch.mm(a, b)
print(a)
print(b)
print(c)
print(d)

元素对应相乘：直接用*就可以，也可以用 torch.mul(a, b) 

矩阵相乘：torch.mm(a,b)此方法只适用于2维矩阵、 torch.matmul(a, b)  推荐使用此方法

一维张量的点乘：torch.dot(a,b)是对两个为1D张量进行点积运算

关于Pytorch的乘法是怎样的问题的解答就分享到这里了，希望以上内容可以对大家有一定的帮助，如果你还有很多疑惑没有解开，可以关注亿速云行业资讯频道了解更多相关知识。