发布时间:2023-05-09 10:00
(一)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。
(五)测试与查看网络结构