当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch Module.modules用法及代码示例


本文简要介绍python语言中 torch.nn.Module.modules 的用法。

用法:

modules()

生成(Yield)

Module - 网络中的一个模块

返回网络中所有模块的迭代器。

注意

重复的模块只返回一次。在以下示例中,l 将仅返回一次。

例子:

>>> l = nn.Linear(2, 2)
>>> net = nn.Sequential(l, l)
>>> for idx, m in enumerate(net.modules()):
        print(idx, '->', m)

0 -> Sequential(
  (0): Linear(in_features=2, out_features=2, bias=True)
  (1): Linear(in_features=2, out_features=2, bias=True)
)
1 -> Linear(in_features=2, out_features=2, bias=True)

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.Module.modules。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。