概述
说明：
这里的tensor不涉及高维张量，即为一维张量或二维张量。
1、a*b
1.1、a和b同行同列
张量中各元素对应相乘。
a = torch.tensor([[1,2],[2,3],[3,4]])
b = torch.tensor([[1,2],[2,3],[3,4]])
print(a*b)
输出：
tensor([[ 1, 4],
[ 4, 9],
[ 9, 16]])
2、torch.mm(a,b)
数学中的矩阵相乘。
a = torch.tensor([[1,2],[2,3],[3,4]])
b = torch.tensor([[1,2,3,4],[1,2,3,4]])
print(torch.mm(a,b))
输出：
tensor([[ 3, 6, 9, 12],
[ 5, 10, 15, 20],
[ 7, 14, 21, 28]])
最后
以上就是端庄百褶裙为你收集整理的torch.tensor的乘法运算的全部内容，希望文章能够帮你解决torch.tensor的乘法运算所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复