Pytorch快速搭建神經網絡

(一)init和forward

每一個torch模塊當中都有__init__()和forward()

__init__()裏面包含搭建神經網絡那些層所需要的信息;

forward()包含神經網絡前向傳遞的內容;比如init裏面的那些層的組合就是由forward來完成;

(二)要繼承父類的__init__()

(三)搭建層

其中對於hidden:n_features是輸入;n_hidden是該層的輸出

同理,對於predict也是同樣的道理;

注意__init__()只是定義了層的信息;而真正搭建是在forward中實現的。

(四)搭建網絡forward

其中從forward中傳來的x要先送入init中定義的hidden,然後再用激勵函數relu激活後輸出,也就是init中hidden的輸出n_hidden,賦值給x;

然後有作爲predict的輸入送入該層,輸出x,賦值給x。

(五)測試與查看網絡結構

 

 

 

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