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())
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值