PyTorch apply

Una funzione che applica una data funzione a tutti i sottomoduli di un nn.Module, in modo simile a map.

Attraversa i moduli con un postorder traversal, dando priorità al figlio sinistro.

1
def do_something(m):
2
  # do something!
3
  return m
4
model = #something very complex model
5
result = model.apply(print_module)