1.前言

Torch 中提供了很多方便的途径, 同样是神经网络, 能快则快, 我们看看如何用更简单的方式搭建同样的神经网络

2.常规方法

class NetWork(nn.Module):def __init__(self,n_input,n_hidden,n_output):super(NetWork,self).__init__()self.hidden = nn.Linear(n_input, n_hidden)self.output_for_predict = nn.Linear(n_hidden, n_output)def forward(self, x):x = F.relu(self.hidden(x))x = self.output_for_predict(x)return x
network = NetWork(n_input = 1,n_hidden = 8, n_output = 2)   #对于二分类问题n_output为2
print(network)  #打印网络结构

3.用nn.Sequential搭建

我们用 class 继承了一个 torch 中的神经网络结构, 然后对其进行了修改, 不过还有更快的方法

network_sequential = nn.Sequential(nn.Linear(1,8),nn.ReLU(),nn.Linear(8,1)
)
print(network_sequential)

4.不同方法对比

我们会发现 network_sequential 相比于network多了ReLU, 在 network 中, 激活函数实际上是在 forward() 功能中才被调用的. 相比network_sequential, network 的好处就是, 你可以根据你的个人需要更加个性化你自己的前向传播过程, 比如(RNN). 不过如果你不需要七七八八的过程, 相信 network_sequential 这种形式更适合你

Pytorch——用nn.Sequential搭建简单的神经网络相关推荐

PyTorch学习笔记（10）--搭建简单的神经网络以及Sequential的使用
PyTorch学习笔记(10)–搭建简单的神经网络以及Sequential的使用本博文是PyTorch的学习笔记,第10次内容记录,主要搭建一个简单的神经网络,并介绍Sequential的使 ...
基于PyTorch，如何构建一个简单的神经网络
本文为 PyTorch 官方教程中:如何构建神经网络.基于 PyTorch 专门构建神经网络的子模块 torch.nn 构建一个简单的神经网络. 完整教程运行 codelab→ https://ope ...
Pytorch:手把手教你搭建简单的卷积神经网络(CNN)，实现MNIST数据集分类任务
关于一些代码里的解释,可以看我上一篇发布的文章,里面有很详细的介绍!!! 可以依次把下面的代码段合在一起运行,也可以通过jupyter notebook分次运行第一步:基本库的导入 import n ...
pytorch torch.nn.Sequential(* args)（嘎哈用的？构建神经网络用的？）
class torch.nn.Sequential(* args) 一个时序容器.Modules 会以他们传入的顺序被添加到容器中.当然,也可以传入一个OrderedDict. 为了更容易的理解如何使 ...
pytorch：nn.Sequential讲解
接下来想讲一下参数初始化方式对训练的影响,但是必须要涉及到pytorch的自定义参数初始化,然而参数初始化又包括在不同结构定义中初始化方式,因而先讲一下pytorch中的nn.Sequential n ...
pytorch中nn.Sequential
一个有序的容器,神经网络模块将按照在传入构造器的顺序依次被添加到计算图中执行,同时以神经网络模块为元素的有序字典也可以作为传入参数. # Example of using Sequentialmode ...
nn.Sequential()
nn.Sequential nn.Sequential()介绍 Pytorch官网举例 nn.Sequential()的本质作用 nn.Sequential()源码 nn.Sequential()介绍 ...
pytorch入门（二）：常见神经网络的原理及实现
什么是卷积神经网络卷积神经网络在图片和语言识别上能给出优秀的结果,近些年被广泛传播和应用.卷积层也叫过滤器,就像上面放置的小灯.(卷积核,滤波器) 我们需要分开来理解: 卷积:我们不对像素进行处理, ...
pytorch 学习笔记：nn.Sequential构造神经网络
1 两种等价的构造方式 1.1 构建Net类: 1.2 使用torch.nn.Sequential nn.Sequential里面的模块按照顺序进行排列的,所以必须确保前一个模块的输出大小和下一个模块 ...

Pytorch——用nn.Sequential搭建简单的神经网络

文章目录

1.前言

2.常规方法

3.用nn.Sequential搭建

4.不同方法对比

Pytorch——用nn.Sequential搭建简单的神经网络相关推荐

最新文章

热门文章