Pytorch笔记3 :类

时间:2025-04-15 07:58:32

()类

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

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

()类的主要属性

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