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

torch.squeeze() 這個函數主要對數據的維度進行壓縮，去掉維數爲1的的維度，默認是將a中全部爲1的維度刪掉。也能夠經過dim指定位置，刪掉指定位置的維數爲1的維度。python torch.unsqueeze()這個函數主要是對數據維度進行擴充。須要經過dim指定位置，給指定位置加上維數爲1的維度。函數 我本身test的代碼：code import torch x = torch.z

>>阅读原文<<