pytorch unsequeeze

`unsqueeze` 是 PyTorch 中的一个函数，用于在指定维度上增加一个维度。具体来说，它会在指定维度上增加一个大小为 1 的维度。例如，如果一个张量的形状为 (3, 4)，那么在第 0 维上使用 `unsqueeze` 后，它的形状将变为 (1, 3, 4)。

下面是一个使用 `unsqueeze` 的例子：

import torch

x = torch.randn(3, 4)
print(x.shape)  # 输出：torch.Size([3, 4])

y = x.unsqueeze(0)
print(y.shape)  # 输出：torch.Size([1, 3, 4])

在这个例子中，我们首先创建了一个形状为 (3, 4) 的张量 `x`，然后在第 0 维上使用了 `unsqueeze`，得到了一个形状为 (1, 3, 4) 的张量 `y`。