U
    zhv                     @   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   H/var/www/html/venv/lib/python3.8/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   