§
    ŠŠtjÏ  ã                   ór   — d dl mZmZmZ d dlmZ d dlmZmZm	Z	m
Z
mZmZmZmZ d dlmZ d dlmZmZ g d¢ZdS )é    )ÚgradÚgrad_and_valueÚvmap)Úreplace_all_batch_norm_modules_)Údebug_unwrapÚfunctionalizeÚhessianÚjacfwdÚjacrevÚjvpÚ	linearizeÚvjp)Ú	rearrange)Úfunctional_callÚstack_module_state)r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   N)Útorch._functorch.apisr   r   r   Ú'torch._functorch.batch_norm_replacementr   Ú!torch._functorch.eager_transformsr   r   r	   r
   r   r   r   r   Útorch._functorch.einopsr   Ú torch._functorch.functional_callr   r   Ú__all__© ó    úQ/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/func/__init__.pyú<module>r      sÚ   ðØ <Ð <Ð <Ð <Ð <Ð <Ð <Ð <Ð <Ð <Ø SÐ SÐ SÐ SÐ SÐ Sð	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð .Ð -Ð -Ð -Ð -Ð -Ø PÐ PÐ PÐ PÐ PÐ PÐ PÐ Pðð ð €€€r   