关于torch.unsqueeze与torch.squeeze

torch内置的unsqueeze方法是增加1的维度,如下

squeeze和unsqueeze相反，是删除数据中1的维度

当指定数据时，删除指定位置的1的维度，当不指定参数时，删除数据中所有1的维度

也有另一种方法可以增加或删除维度，如下

 也可以指定位置

 使用这种方法如果想删除维度可以采用切片的方式将其取出，也可以使用squeeze方法。