torch.unsqueeze() 返回在指定位置插入的尺寸为1的新张量 dim-插入的单个尺寸的索引 import torch X=torch.tensor([1,2,3,4]) X.size() torch.Size([4]) X1=torch.unsqueeze(X,0) # 在第0维上增加“1”,在第一个维度上增加一个维度 X1 tensor([[1, 2, 3, 4]]) X1.size() torch.Size([1, 4]) 参考博客