(一)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。
(五)測試與查看網絡結構