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]
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]