Pytorch快速搭建神经网络

发布时间:2023-05-09 10:00

(一)init和forward

每一个torch模块当中都有__init__()和forward()

Pytorch快速搭建神经网络_第1张图片

__init__()里面包含搭建神经网络那些层所需要的信息;

forward()包含神经网络前向传递的内容;比如init里面的那些层的组合就是由forward来完成;

(二)要继承父类的__init__()

(三)搭建层

其中对于hidden:n_features是输入;n_hidden是该层的输出

同理,对于predict也是同样的道理;

注意__init__()只是定义了层的信息;而真正搭建是在forward中实现的。

Pytorch快速搭建神经网络_第2张图片

(四)搭建网络forward

其中从forward中传来的x要先送入init中定义的hidden,然后再用激励函数relu激活后输出,也就是init中hidden的输出n_hidden,赋值给x;

然后有作为predict的输入送入该层,输出x,赋值给x。

Pytorch快速搭建神经网络_第3张图片

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

Pytorch快速搭建神经网络_第4张图片

 

 

 

ItVuer - 免责声明 - 关于我们 - 联系我们

本网站信息来源于互联网,如有侵权请联系:561261067@qq.com

桂ICP备16001015号