pytorch中的神经网络模块基础类——torch.nn.Module

1.torch.nn.Module概要

pytorch官网对torch.nn.Module的描述如下。

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

2.torch.nn.Module.parameters

pytorch官网对torch.nn.Module.parameters的描述如下：该参数返回的是当前神经网络模块的所有参数的迭代器。若recuse参数为True，则返回的迭代器包含了当前模块的子模块。反之则不包含。

如下述pytorch官网代码所示，recuse参数默认为True。从源码中的yield参数进一步说明了torch.nn.Module.parameters返回的是一个迭代器。