
    קgw                     &    d dl mZ ej        Zd ZdS )    )dimc                 V     t          |          \  }} | fd|D                       S )Nc              3   .   K   | ]} |          V  d S )N ).0vfns     R/var/www/html/ai-engine/env/lib/python3.11/site-packages/functorch/dim/tree_map.py	<genexpr>ztree_map.<locals>.<genexpr>   s+      ''qRRUU''''''    )tree_flatten)r	   treevs	unflattens   `   r
   tree_mapr      s9     &&MB	9''''B''''''r   N)functorch._Cr   r   r   r   r   r
   <module>r      s<          ( ( ( ( (r   