關於 pytorch inplace operation須要注意的問題(data和detach方法的區別)

(本文章適用於 pytorch0.4.0 版本, 既然 Variable 和 Tensor merge 到一塊了, 那就叫 Tensor吧)html 在編寫 pytorch 代碼的時候, 若是模型很複雜, 代碼寫的很隨意, 那麼頗有可能就會碰到由 inplace operation 致使的問題. 因此本文將對 pytorch 的 inplace operation 作一個簡單的總結。python
相關文章
相關標籤/搜索