torch.squeeze

torch.squeeze函数用于从张量中移除尺寸为1的维度。它可以减少张量的维度，使得张量更紧凑。

具体来说，torch.squeeze可以有两种使用方式：

1. 使用torch.squeeze(tensor)：这种用法会移除张量tensor中所有尺寸为1的维度。例如，对于形状为(1, 3, 1, 5)的张量，使用torch.squeeze(tensor)后，它将变为形状为(3, 5)的张量。

2. 使用torch.squeeze(tensor, dim)：这种用法会只移除指定维度dim上尺寸为1的维度。例如，对于形状为(1, 3, 1, 5)的张量，使用torch.squeeze(tensor, 0)后，它将变为形状为(3, 1, 5)的张量，而使用torch.squeeze(tensor, 2)后，它将变为形状为(1, 3, 5)的张量。

需要注意的是，torch.squeeze不会改变张量的数据，只是改变了张量的维度。如果要在原地修改张量，请使用tensor.squeeze_()方法。