一文掌握torch.squeeze() 和torch.unsqueeze()的用法

torch.squeeze() 這個函數主要對數據的維度進行壓縮，去掉維數爲1的的維度，默認是將a中所有爲1的維度刪掉。也可以通過dim指定位置，刪掉指定位置的維數爲1的維度。 torch.unsqueeze()這個函數主要是對數據維度進行擴充。需要通過dim指定位置，給指定位置加上維數爲1的維度。 我自己test的代碼： import torch x = torch.zeros(3,2,4,1

>>阅读原文<<