关于pytorch不区分行向量与列向量的实验

关于pytorch不区分行向量与列向量的实验,第1张

关于pytorch不区分向量与列向量的实验

听李沐老师讲深度学习时候解释pytorch不区分行向量和列向量,只相当于是一维数组,为了理解透彻,做了一些实验

r=torch.tensor([1,2,3],dtype=torch.float32)
print(r,r.T)


说明确实行和列向量没有区分
再看看在矩阵与向量乘法中向量的角色

r=torch.tensor([1,2,3],dtype=torch.float32)
print(r,r.T)
m=torch.ones(3,3)
print(torch.mv(m,r))
print(torch.mv(r.T,m))


可以看出,在乘法中矩阵右乘向量可以得到正确结果,向量乘矩阵报错,因此在pytorch中还是倾向于将向量理解成列向量比较合适。

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

原文地址: https://outofmemory.cn/zaji/5711856.html

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

发表评论

登录后才能评论

评论列表(0条)

保存