Pytorch unsqueeze()函数的用法 在这里插入代码片 import torch a=torch.ones(10) print("a.shape:",a.shape) b=a.unsqueeze(0) #对0维度扩展一维 c=a.unsqueeze(1) #对一维度扩展一维 print("b.shape",b.shape) print("c.shape",c.shape) 运行结果