分类：unsqueeze

2021
02-21

pytorch下的unsqueeze和squeeze的用法说明

#squeeze函数：从数组的形状中删除单维度条目，即把shape中为1的维度去掉#unsqueeze()是squeeze()的反向操作，增加一个维度，该维度维数为1，可以指定添加的维度。例如unsqueeze(a,1)表示在1这个维度进行添加importtorcha=torch.rand(2,3,1)print(torch.unsqueeze(a,2).size())#torch.Size([2,3,1,1])print(a.size())#torch.Size([2,3,1])print(a.squeeze().size())#torch.Size([2,3])print(a.squ... 继续阅读 >