PyTorch中的torch.Tensor.view()和torch.Tensor.view

PyTorch中的torch.Tensor.view()和torch.Tensor.view,第1张

PyTorch中的torch.Tensor.view()和torch.Tensor.view 1、torch.Tensor.view()

相当于 Numpy.reshape() *** 作,view() 方法返回一个新的 tensor,如果用新的 dtype 对 tensor 进行变换,新的 dtype 的数据类型字节数(比如64、32或16)必须与变换前的 dtype 的数据类型字节数相同。

2、torch.Tensor.view_as()

torch.Tensor.view_as(other) → Tensor

返回的 tensor 与 other 的 size 相同,即构造一个与 other 相同维度的新 tensor。相当于 self.view(other.size()) 的效果。

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

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

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

发表评论

登录后才能评论

评论列表(0条)

保存