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。

(五)测试与查看网络结构

 

 

 

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