torch.squeeze()函数的理解

import torch
a=torch.Tensor([[1],[1]])
b=torch.squeeze(a)
c=1
print(' a :', a ,' b: ', b , ' c: ' , c)
print('\n',a.size(),b.size())

我们可以看到本来2x1=>2的维度，也就是，[[1],[1]]=>[1,1]