o hw@sddlmZejZddZdS))dimcs"t|\}}|fdd|DS)Nc3s|]}|VqdS)N).0vfnrj/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/functorch/dim/tree_map.py sztree_map..) tree_flatten)rtreevs unflattenrrrtree_map s rN) functorch._Crr rrrrrs