神经网络系列会持续学习并整理
一、前言
作为新手小白,在入门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