1. torch.squeeze(input, dim=None, out=None)
input是输入的参数,dim是指定要合并维度为1的所在维度
当dim=0时原样输出,当dim=1时该维度上数值为1则合并,否者不合并,其他维度上同理。
2. torch.unsqueeze(input, dim, out=None)
插入一个维度唯一的维度
dim=0原样输出,dim=1在山上插入维度为1 的维度,dim=2在列上插入维度为1 的维度
比如某一tensor为(2,4)
当dim=1时变成(2,1,4)两个1行4列的tensor
当dim=2时变成(2,4,1)变成两个4行1列的tensor