pytorch

torch.squeeze() 這個函數主要對數據的維度進行壓縮,去掉維數爲1的的維度,設圖像大小爲(1,36,60),可以去掉,變爲(36,60)

torch.fromnumpy()

 

torch.unsqueeze()都數據進行擴充(36,60),擴充爲(1,36,60)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章