PyTorch 代码详解：squeeze 函数的使用

在 PyTorch 中，squeeze 函数可以用于去除输入数据中的维度为 1 的维度。在神经网络模型中，我们经常需要将输入数据的维度进行调整。例如，以下代码展示了如何使用 squeeze 函数将输入数据的维度从 4 维降到 3 维：

def forward(self, x):
    x = torch.squeeze(x, dim=3)  # 将输入数据的维度从4维降到3维
    x = self.conv1(x)

代码中，torch.squeeze(x, dim=3) 将输入数据 x 的第 3 维度去除，并将其维度从 4 维降到 3 维。

注意：

• squeeze 函数可以去除任何维度为 1 的维度，只需要指定 dim 参数即可。
• 如果数据中不存在维度为 1 的维度，squeeze 函数将不会改变数据的维度。

希望本文能够帮助您理解 PyTorch 中 squeeze 函数的使用。