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

2020-02-12  本文已影响0人  yuanCruise
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返回的是一个迭代器。


上一篇 下一篇

猜你喜欢

热点阅读