pytorch快速搭建神经网络_Sequential

你的名字 2021-05-12 11:49 582阅读 0赞

#### 之前用`Class类`来搭建神经网络 ####

class Neuro_net(torch.nn.Module):
        """神经网络"""
        def __init__(self, n_feature, n_hidden_layer, n_output):
            super(Neuro_net, self).__init__()
            self.hidden_layer = torch.nn.Linear(n_feature, n_hidden_layer)
            self.output_layer = torch.nn.Linear(n_hidden_layer, n_output)
    
        def forward(self, input):
            hidden_out = torch.relu(self.hidden_layer(input))
            out = self.output_layer(hidden_out)
            return out
        
        
    net = Neuro_net(2, 10, 2)
    print(net)

##### class类图结构： #####

![在这里插入图片描述][20200323170106633.png]

### 使用`torch.nn.Sequential()` 快速搭建神经网络 ###

net = torch.nn.Sequential(
        torch.nn.Linear(2, 10),
        torch.nn.ReLU(),
        torch.nn.Linear(10, 2)
    )
    print(net)

##### Sequential图结构 #####

![在这里插入图片描述][20200323170113796.png]

### 总结： ###

*  我们可以发现，使用`torch.nn.Sequential`会自动加入激励函数, 但是 class类net 中, 激励函数实际上是在 forward() 功能中才被调用的
 *  使用class类中的`torch.nn.Module`，我们可以根据自己的需求改变传播过程
 *  如果你需要快速构建或者不需要过多的过程，直接使用`torch.nn.Sequential`吧

[20200323170106633.png]: /images/20210511/7f2f4e796e1f4ab9845766098ecd6746.png
[20200323170113796.png]: /images/20210511/9fda3c506f734f11856f51ecbb702f72.png

发表评论取消回复

表情：

评论列表（有 0 条评论，582人围观）

还没有评论，来说两句吧...

相关阅读

相关 python编程搭建cnn神经网络

使用Python编程来搭建CNN神经网络可以使用TensorFlow或Keras框架。TensorFlow提供了几个高级API，比如tf.keras，使得构建CNN神经网络变得

超、凢脫俗/ 2024年03月25日 11:11/ 0 赞/ 23 阅读

相关迁移学习之快速搭建【卷积神经网络】

前言卷积神经网络概念认识：[一篇文章“简单”认识《卷积神经网络》(更新版)][Link 1] 卷积神经网络简单模型搭建：[手把手搭建一个【卷积神经网络】][Lin

阳光穿透心脏的1/2处/ 2023年01月20日 10:59/ 0 赞/ 60 阅读

相关 pytorch搭建神经网络过程

文章目录搭建网络基本流程数据集整理数据集分类:训练集验证集测试机数据集读取数据增加和数据标

曾经终败给现在/ 2022年10月05日 05:53/ 0 赞/ 202 阅读

相关【Pytorch学习】用pytorch搭建第一个神经网络

简述虽然使用过pytorch来搭建GANs，但是对于Pytorch其实还是不是很熟，这里想系统的学习下。顺便再来做一下笔记。学习网站 > [https://m

浅浅的花香味﹌/ 2022年04月18日 05:24/ 0 赞/ 304 阅读

相关搭建神经网络的基本步骤

1 搭建神经网络的步骤搭建神经网络，可分为四步完成：准备工作、前向传播、反向传播、循环迭代 1-0 准备工作导入模块，生成数据集 > import > >

末蓝、/ 2022年03月10日 00:54/ 0 赞/ 653 阅读

相关快速入门PyTorch(2)--如何构建一个神经网络

> 原文链接：[mp.weixin.qq.com/s/Q8tNXsDh6…][mp.weixin.qq.com_s_Q8tNXsDh6] 快速入门 PyTorch 教程第二篇

川长思鸟来/ 2022年01月18日 23:47/ 0 赞/ 248 阅读

相关 pytorch搭建神经网络

-- coding: utf-8 -- @Time :2019/6/30 21:07 @Author :XiaoMa import t

向右看齐/ 2021年12月19日 21:21/ 0 赞/ 288 阅读

相关 pytorch快速搭建神经网络_Sequential

之前用`Class类`来搭建神经网络 class Neuro_net(torch.nn.Module): """神经网络""" d

你的名字/ 2021年05月12日 11:49/ 0 赞/ 583 阅读

相关 pytorch_搭建简单神经网络（分类）

用pytorch搭建神经网络，将数据分成两类环境 pytoch1.1.0 1. 生成训练数据准备数据 data = torch.one

你的名字/ 2021年05月12日 11:49/ 0 赞/ 651 阅读

相关 pytorch—搭建简单神经网络（回归）

用pytorch搭建神经网络，拟合曲线： y = x 2 y=x^2 y=x2 环境 pytoch1.1.0 1. 生成训练数据准备数据

红太狼/ 2021年05月12日 11:49/ 0 赞/ 575 阅读