pytorch squeeze()和unsqueeze()

squeeze()： squeeze(arg)表示第arg維的維度值爲1，則去掉該維度。否則tensor不變。（即若tensor.shape()[arg] = 1，則去掉該維度） unsqueeze()： unsqueeze(arg)與squeeze(arg)作用相反，表示在第arg維增加一個維度值爲1的維度。 常用操作： 總結： squeeze和unsqueeze都不會增刪數據本身，只是改變矩陣

>>阅读原文<<