tensordict.nn.get_functional

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

将 nn.Module 原地转换为函数式模块，并返回此模块的状态版本，该版本可在函数式设置中使用。