pytorch 增加一个维度

在 PyTorch 中，可以使用 `unsqueeze()` 方法在张量中增加一个维度。这个方法需要一个参数，即要增加的维度的索引，从 0 开始计数。例如，如果要在形状为 (3, 4) 的张量中增加一个维度，可以这样做：

import torch

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

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

这个方法将在索引为 0 的位置增加一个维度，使得原来的形状 (3, 4) 变成了 (1, 3, 4)。如果要在其他位置增加维度，只需要相应地修改参数即可。