pytorch中的神经网络模块基础类——torch.nn.Mod
2020-02-12 本文已影响0人
yuanCruise
1.torch.nn.Module概要
pytorch官网对torch.nn.Module的描述如下。
![](https://img.haomeiwen.com/i5529997/25c866cf6bfc1d7c.png)
torch.nn.Module是所有的神经网络模块的基类,且所有的神经网络模块都可以包含其他的子神经网络模块,这些子神经网络模块可以作为类的属性赋值。
2.torch.nn.Module.parameters
pytorch官网对torch.nn.Module.parameters的描述如下:该参数返回的是当前神经网络模块的所有参数的迭代器。若recuse参数为True,则返回的迭代器包含了当前模块的子模块。反之则不包含。
![](https://img.haomeiwen.com/i5529997/ac90263b56940c4b.png)
如下述pytorch官网代码所示,recuse参数默认为True。从源码中的yield参数进一步说明了torch.nn.Module.parameters返回的是一个迭代器。
![](https://img.haomeiwen.com/i5529997/6c43384f9b3ec3ed.png)