torch.func.stack\_module\_state =============================== .. currentmodule:: torch.func .. autofunction:: stack_module_state