Python PyTorch ScriptModule.modules用法及代碼示例

本文簡要介紹python語言中 torch.jit.ScriptModule.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)