unsqueeze(dim=1)

unsqueeze(dim=1)是PyTorch中的一个方法，用于在张量中在指定的位置（维度）上添加一个大小为1的新维度。这个方法可以改变张量的形状，使其适应特定的计算需求。

以下是一个演示unsqueeze(dim=1)的例子：

import torch

# 创建一个2维张量
tensor = torch.tensor([[1, 2, 3],
                       [4, 5, 6]])

# 在第1个维度上添加一个新维度
new_tensor = tensor.unsqueeze(dim=1)

print("原始张量形状：", tensor.shape)  # 输出：(2, 3)
print("添加新维度后的张量形状：", new_tensor.shape)  # 输出：(2, 1, 3)

在这个例子中，我们创建了一个2维张量tensor，形状为(2, 3)。然后，我们使用unsqueeze(dim=1)在第1个维度上添加了一个新维度，得到了形状为(2, 1, 3)的new_tensor。