torch.unsqueeze 是什么意思

torch.unsqueeze 是 PyTorch 中的一个函数，用于将张量的维度增加一维。它的作用是在指定的维度上插入一个大小为1的维度，从而扩展张量的形状。例如，如果一个张量的形状为 (3, 4)，则将其在维度0上增加一维，形状变为 (1, 3, 4)。具体用法如下：

import torch

x = torch.randn(3, 4)
y = torch.unsqueeze(x, 0)
print(y.shape)  # 输出 (1, 3, 4)

在这个例子中，我们将张量 `x` 在维度0上增加了一维，得到了新的张量 `y`。