pytorch squeeze()

`torch.squeeze()` 是 PyTorch 中的一个函数，用于减少维度。它会移除张量中维度大小为1的维度。假设张量的形状为 `(1, 2, 1, 3)`，使用 `squeeze()` 后，形状将变为 `(2, 3)`。如果在调用 `squeeze()` 时指定了一个维度参数，则只会移除指定维度大小为1的维度。例如，对于形状为 `(1, 2, 1, 3)` 的张量，可以使用 `squeeze(dim=2)` 来移除第2个维度大小为1的维度，结果将变为 `(1, 2, 3)`。

需要注意的是，`squeeze()` 并不会改变张量的内容，只是改变了其形状。如果希望在指定维度上进行减少维度操作并且改变张量的内容，可以使用 `unsqueeze()` 函数进行扩展维度或者使用其他相关函数进行操作。