介绍:view是pytorch中tensor的一种方法
tensor.view(shape)
功能:将tensor变成另外另外一个大小与shape相同的tensor
实例:
>>> x = torch.randn(4, 4)
>>> x.size()
torch.Size([4, 4])
>>> y = x.view(16)
>>> y.size()
torch.Size([16])
特别的当shape中有某个维数为-1时,则该维的维数会根据其他维的维数自动确定
实例:
>>> z = x.view(-1, 8)
>>> z.size()
torch.Size([2, 8]) #2*8=16 所以-1的维数为2