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