神经网络的基本骨架-nn.module的使用

神经网络系列会持续学习并整理

---

一、前言

作为新手小白，在入门pytorch时，要想明白神经网络，我觉得得先知道神经网络nn.module的基本骨架。以下内容是我参照土堆老师，写的代码及注释，初步对神经网络架构有了一点了解。神经网络的基本骨架-nn.Module的使用_哔哩哔哩_bilibili

二、初步理解框架

1.图解

输入、输出、前项通道之间的关系，如下图所示。

2.代码及注释

代码如下（示例）：

import torch
from torch import nn

class Tudui(nn.Module):      #创建Tudui这个类
    def __init__(self):      #初始化
        super(Tudui, self).__init__()

    def forward(self,input): #前项通道
        output=input+1
        return output

tudui=Tudui()            #调用Tudui
x=torch.tensor(1.0)      #给上一行定义x输入
output=tudui(x)          #完成输出output
print(output)            #打印output

---

 

三、总结

以上是我对神经网络的理解，接下来会继续学习并整理关于神经网络的一些内容，视频教程神经网络的基本骨架-nn.Module的使用_哔哩哔哩_bilibili