
    sgw                     .    d dl mZ ej                  Zd Zy)    )dimc                 B     t        |      \  }} | fd|D              S )Nc              3   .   K   | ]  } |        y w)N ).0vfns     I/var/www/html/venv/lib/python3.12/site-packages/functorch/dim/tree_map.py	<genexpr>ztree_map.<locals>.<genexpr>   s     'qRU's   )tree_flatten)r	   treevs	unflattens   `   r
   tree_mapr      s"     &MB	'B'''    N)functorch._Cr   r   r   r   r   r
   <module>r      s     (r   