以下是PyTorch文档中的描述: torch.squeeze(input, dim=None, *, out=None)→ Tensor 返回一个张量,将所有大小为1的输入维度都移除。 例如,如果输入的形状为(A×1×B×C×1×D),则输出张量的形状为(A×B×C×D)。 当给定参数dim时,只在给定的维度上进行挤压操作。如果input的形状为(A×1×B),squeez...