torch 中的torch.squeeze()和torch.unsqueeze()

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