tensor增加维度

可以使用PyTorch中的unsqueeze()函数来增加维度。unsqueeze()函数可以在指定位置增加一个维度。例如，如果有一个大小为（2,3）的张量，我们可以使用unsqueeze()函数在第一个位置上增加一个维度，生成一个大小为（1,2,3）的张量。

示例代码如下：

import torch

# 创建一个大小为（2,3）的张量
x = torch.randn(2, 3)
print(x.size())

# 在第一个位置上增加一个维度
x = x.unsqueeze(0)
print(x.size())

输出结果为：

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

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