相当于 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()) 的效果。
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)