下。unsqueeze

unsqueeze是PyTorch中的一个函数，用于在指定维度上增加维度大小为1的维度。它的作用是将原来的张量进行扩展，使得原来的维度上的元素个数变为1。unsqueeze函数的用法如下：

import torch

# 创建一个3x3的张量
x = torch.tensor([[1, 2, 3],
                  [4, 5, 6],
                  [7, 8, 9]])

# 在第0维度上增加一个维度
y = torch.unsqueeze(x, 0)
print(y.shape)  # 输出：torch.Size([1, 3, 3])

# 在第1维度上增加一个维度
z = torch.unsqueeze(x, 1)
print(z.shape)  # 输出：torch.Size([3, 1, 3])

在上面的例子中，我们创建了一个3x3的张量x，并使用unsqueeze函数在第0维度和第1维度上增加了一个维度。可以看到，通过unsqueeze函数，我们可以在指定的维度上增加维度大小为1的维度。