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返回的是一个迭代器。