如何增加tensor第三个维度

要增加一个 tensor 的第三个维度，可以使用 PyTorch 中的 unsqueeze() 函数。这个函数可以在指定的维度上增加一个维度，例如：

import torch

# 创建一个 2x3 的二维 tensor
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 在第三个维度上增加一个维度
x = x.unsqueeze(2)

# 查看新的 tensor 的形状
print(x.shape)

输出结果为：

torch.Size([2, 3, 1])

这样就在第三个维度上增加了一个维度。