קg<ddlmZddlmZmZdaedZdS))contextmanager)_vmap_add_layers_vmap_remove_layersFc#KtrJtd|D}t|} t|dadVdat |dS#dat |wxYw)Nc3\K|]'}t|t|j|jfV(dS)N) isinstanceint_levelsize).0ds V/var/www/html/ai-engine/env/lib/python3.11/site-packages/functorch/dim/batch_tensor.py z!_enable_layers..s:NN!:a;M;MNAHaf%NNNNNNTF)_enabledsortedlenrr)dimsinputns r_enable_layersrs NNtNNN N NE E A AAs AA0N) contextlibrtorch._C._functorchrrrrrrrs] &%%%%%EEEEEEEE      r