squeeze（）
squeeze 函数是 PyTorch 中的一个函数，用于从张量（Tensor）中去除所有长度为 1 的维度。这在处理神经网络模型的输出时非常有用，因为模型的输出可能包含一些不必要的单一维度，例如在批处理大小为 1 的情况下，输出可能会有一个额外的批次维度。
简单来说就是从张量（Tensor）中去除所有长度为 1 的维度。
实例：
import torch
x = torch.randn(1, 2, 1, 2, 2)
print(x.shape) # 输出: torch.Size([1, 2, 1, 2, 2])
y = torch.squeeze(x)
print(y.shape) # 输出: torch.Size([2, 2, 2])
z = torch.squeeze(x, dim=1)
print(z.shape) # 输出: torch.Size([1, 2, 1, 2, 2])
v= torch.squeeze(x, dim=0)
print(v.shape) # 输出: torch.Size([1, 2, 2, 2])输出：
#x：
torch.Size([1, 2, 1, 2, 2])
#y：不指定维度就去除所有长度为 1 的维度
torch.Size([2, 2, 2])#z：指定维度长度不为1则无效
torch.Size([1, 2, 1, 2, 2])#v：指定维度长度为1生效
torch.Size([2, 1, 2, 2])