x.view[-1,参数a]以及x.view[参数a,-1]语句的作用就是根据参数来调整维度
import torch
a = torch.rand(2,4,4)
a = a.view(-1,2)
print(a.shape)
上段代码的意思就是调整为2列的维度,因为一共32个元素,所以最后应该是(16,2)的tensor
torch.Size([16, 2])
同理,如果是x.view[参数,-1]就是根据参数来调整列数
a = torch.rand(2,4,4)
a = a.view(2,-1)
print(a.shape)
--结果 torch.Size([2, 16])
x.view[-1]因为行数,列数参数都未知,最后就调整为一维tensor
a = torch.rand(2,4,4)
a = a.view(-1)
print(a.shape)
--结果 torch.Size([32])