【Pytorch】tensor类型数据.squeeze()和.unsqueeze()函数的简明教程(一看就会)
文章目录1squeeze1.11.21.3执行操作后需要写回2unsqueezepytorch系列代码中常见的两个函数squeeze()和unsqueeze()1squeeze1.1importtorchx=torch.randn(2,1,2)#生成一个维度是(2,1,2)的tensor向量print(x.shape)#torch.Size([2,1,2])x=x.squeeze()#参数为空,表