PyTorch中的view函数可以改变tensor的形状

eg1

import torch
test = torch.zeros(2,3)
print(test)
print(test.view(3,-1)) #-1表示程序会根据上一个维度自动计算-1所在位置的维度

#tensor([[0., 0., 0.],
#        [0., 0., 0.]])

#tensor([[0., 0.],
#        [0., 0.],
#        [0., 0.]])

eg2

test = torch.Tensor([1,2,3,4,5,6,7,8])
print(test)
print(test.view(4,2))

# tensor([1., 2., 3., 4., 5., 6., 7., 8.])

#tensor([[1., 2.],
#        [3., 4.],
#        [5., 6.],
#        [7., 8.]])