oneflow.nn.Module.children

Module.children()Iterator[“Module”]

Returns an iterator over immediate children modules.

Yields

Module – a child module

Example:

>>> import oneflow.nn as nn
>>> l1 = nn.Linear(2, 2)
>>> l2 = nn.Linear(2, 2)
>>> net = nn.Sequential(l1, l2)
>>> for idx, m in enumerate(net.children()):
...     print(idx, '->', m)
0 -> Linear(in_features=2, out_features=2, bias=True)
1 -> Linear(in_features=2, out_features=2, bias=True)