PyTorch中的torch.Tensor.view()和torch.Tensor.view_as()的用法详解

Posted 2021-12-11 魏晓蕾

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()) 的效果。