pytorch 去掉全连接层_在PyTorch中建立神经网络的三种方法

742f0fea3908ff5bd682b4a7b2319398.png

介绍

假设您要定义以下神经网络,其中一个输入层,两个隐藏层和一个输出层,中间层使用relu激活,而输出层使用sigmoid激活函数,如下所示:

1568f543fdeb4175a4cb6d10eed3a49e.png

全连接(前馈)网络

这是一个全连接的16x12x10x1神经网络。

1.手动建立权重和偏差

一种方法是构建所有的块。这是一种低级的方法,但是如果您试图理解深度学习体系结构(或者如果你想开发一个定制层),那么它可能非常适合您。这里的关键是,您需要告诉PyTorch在您的网络中什么是可变的或可优化的,以便PyTorch知道如何在您的网络中执行梯度下降。让我们看看在低层PyTorch中人们是如何处理这个问题的:

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值