o
    I&i                     @   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   AC:\wamp64\www\opt\env\Lib\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   