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)