torch.Tensor.view (Python method, in torch.Tensor)

返回具有相同数据但大小不同的新张量。
返回的张量共享相同的数据,必须具有相同数量的元素,但可能有不同的大小。

Example

>>> x = torch.randn(4, 4)
>>> x.size()
torch.Size([4, 4])
>>> y = x.view(16)
>>> y.size()
torch.Size([16])
>>> z = x.view(-1, 8)  # the size -1 is inferred from other dimensions
>>> z.size()
torch.Size([2, 8])
原文地址:https://www.cnblogs.com/lindaxin/p/7986915.html