Pytorch笔记3 :nn.model（）类

nn.model()类

在学习nn.model()类之前，先简要了解一下nn。
nn主要有四个模块

• nn.Parameter: 一个张量的子类，用于表示可学习的参数 w， b
• nn.Module： 网络层的基类，用于管理网络的属性，LeNet是一个module类，LeNet的子模块例如conv2，也是一个nn.module类
• nn.functional：用于函数的实现，比如卷积运算，加法运算
• nn.__init__：参数初始化方法

nn.Module()类的主要属性

• parameter ： 用于存储和管理Parameter类
• Module ： 用于存储和管理Module类相关
• buffers ：存储缓冲属性，比如均值等
• 其他五个是用于管理钩子函数（_hocks()）