o
    Zhw                     @   s   d dl mZ ejZdd ZdS )    )dimc                    s"   t |\}}| fdd|D S )Nc                 3   s    | ]} |V  qd S )N ).0vfnr   E/var/www/auris/lib/python3.10/site-packages/functorch/dim/tree_map.py	<genexpr>   s    ztree_map.<locals>.<genexpr>)tree_flatten)r   treevsZ	unflattenr   r   r   tree_map   s   r   N)Zfunctorch._Cr   r
   r   r   r   r   r   <module>   s   