ott.neural.networks.velocity_field.mlp.MLP.iter_modules

ott.neural.networks.velocity_field.mlp.MLP.iter_modules#

MLP.iter_modules()#

Warning: this method is method is deprecated; use iter_modules() instead.

Recursively iterates over all nested Module’s of the current Module, including the current Module. Alias of iter_modules().

Return type:

Iterator[tuple[tuple[Key, ...], Module]]