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的乘法是怎样的问题的解答就分享到这里了,希望以上内容可以对大家有一定的帮助,如果你还有很多疑惑没有解开,可以关注亿速云行业资讯频道了解更多相关知识。
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。