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。

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

 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值