()类
在学习()类之前,先简要了解一下nn。
nn主要有四个模块
- : 一个张量的子类,用于表示可学习的参数 w, b
- : 网络层的基类,用于管理网络的属性,LeNet是一个module类,LeNet的子模块例如conv2,也是一个类
- :用于函数的实现,比如卷积运算,加法运算
- nn.__init__:参数初始化方法
()类的主要属性
- parameter : 用于存储和管理Parameter类
- Module : 用于存储和管理Module类相关
- buffers :存储缓冲属性,比如均值等
- 其他五个是用于管理钩子函数(_hocks())