1.torch.nn.Module概要

pytorch官网对torch.nn.Module的描述如下。
pytorch中的神经网络模块基础类——torch.nn.Module

torch.nn.Module是所有的神经网络模块的基类,且所有的神经网络模块都可以包含其他的子神经网络模块,这些子神经网络模块可以作为类的属性赋值。

2.torch.nn.Module.parameters

pytorch官网对torch.nn.Module.parameters的描述如下:该参数返回的是当前神经网络模块的所有参数的迭代器。若recuse参数为True,则返回的迭代器包含了当前模块的子模块。反之则不包含。
pytorch中的神经网络模块基础类——torch.nn.Module
如下述pytorch官网代码所示,recuse参数默认为True。从源码中的yield参数进一步说明了torch.nn.Module.parameters返回的是一个迭代器。
pytorch中的神经网络模块基础类——torch.nn.Module

相关文章:

  • 2022-01-07
  • 2021-04-29
  • 2022-01-23
  • 2021-05-05
  • 2021-05-24
  • 2021-07-18
猜你喜欢
  • 2022-12-23
  • 2021-09-25
  • 2021-07-15
  • 2021-06-28
  • 2021-05-23
  • 2021-06-07
  • 2023-03-23
相关资源
相似解决方案