a
    hw                     @   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   D/var/www/auris/lib/python3.9/site-packages/functorch/dim/tree_map.py	<genexpr>       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   