pytorch增加维度

在 PyTorch 中，可以使用 `unsqueeze()` 函数来增加维度。该函数接受一个整数作为参数，表示在哪个位置增加维度。例如，假设你有一个形状为 `(3, 4)` 的张量 `x`，你想在第二个维度上增加一个维度，可以使用以下代码：

import torch

x = torch.randn(3, 4)
y = x.unsqueeze(1)  # 在第二个维度上增加一个维度

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

在上面的例子中，`unsqueeze()` 函数将张量 `x` 在第二个维度上增加了一个维度，结果张量 `y` 的形状变为 `(3, 1, 4)`。