pytorch 之 __call__, __init__,forward
要學習pytorch咙好,一個前提是 知道python calss中的__call__和__init__方法.
簡單的說就是:
__init__: 類的初始化函數岗宣,類似于c++的構造函數
__call___: 使得類對象具有類似函數的功能菩鲜。(使對象具有函數功能)
__init__比較好理解,現在主要看一下__call__的功能示例:
在調用的時候傳入了參數
發(fā)現對象a的表現完全類似一個函數胳螟。
新生成的對象a不像一個變量昔馋,更像一個函數
在call 里面調用forward函數
先進行對象初始化,之后在對forward里的函數進行運算糖耸,最后返回的是forward里的數值