DŽg`ddlmZmZmZmZmZmZmZddlm Z m Z ddl m Z m Z ddlmZddlmZy))vjpjvpjacrevjacfwdhessian functionalize linearize)gradgrad_and_value)functional_callstack_module_state)replace_all_batch_norm_modules_)vmapN)!torch._functorch.eager_transformsrrrrrrr torch._functorch.apisr r torch._functorch.functional_callr r 'torch._functorch.batch_norm_replacementrr[/home/mcse/projects/flask_80/flask-venv/lib/python3.12/site-packages/torch/func/__init__.pyrs"7PS&r