pytorch Tensor形狀問題

Pytorch Tensor形狀 編寫神經網絡的時候,經常會需要獲取tensor的形狀,使用tensor.shape或tensor.size()即可獲取。 今天調試代碼的時候發現一個問題 在使用pycharm調試兩個程序的過程中,同樣的代碼返回的結果是不一樣的,一個返回的是tuple類型,一個返回的是tenosr。暫時還不知道原因是什麼,先佔個坑記錄一下。
相關文章
相關標籤/搜索