pytorch squeeze()和unsqueeze()

squeeze()： squeeze(arg)表示第arg维的维度值为1，则去掉该维度。否则tensor不变。（即若tensor.shape()[arg] = 1，则去掉该维度） unsqueeze()： unsqueeze(arg)与squeeze(arg)作用相反，表示在第arg维增加一个维度值为1的维度。 常用操作： 总结： squeeze和unsqueeze都不会增删数据本身，只是改变矩阵

>>阅读原文<<