torch.Tensor乘法

torch.Tensor乘法,第1张

torch.Tensor乘法
a = torch.Tensor(np.random.normal(0,1,(2,1,1,1)))
b = torch.Tensor(np.random.normal(0,1,(2,3,4,4)))
c=a*b

c是由a和b点乘得来,先将a扩张成b的形状,数字逐位点乘。

a = torch.Tensor(np.random.normal(0,1,(2,3)))
b = torch.Tensor(np.random.normal(0,1,(3,4)))
c=torch.mm(a,b)

二维矩阵乘法,符合数学运算矩阵乘法,生成形状为(2,4)

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/zaji/5443802.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-12-11
下一篇 2022-12-11

发表评论

登录后才能评论

评论列表(0条)

保存