lsy5557的博客

原创 pytorch中的X.view[-1],X.view[-1,参数],X.view[参数，-1]

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