tensordict.nn.get_functional

tensordict.nn.get_functional(module: nn.Module, funs_to_decorate: Iterable[str] | None = None) → nn.Module

Converts a nn.Module to a functional module in-place, and returns a stateful version of this module that can be used in functional settings.