a Ç Àhœã@s0ddlmZddlmZmZdaedd„ƒZdS)é)Úcontextmanager)Ú_vmap_add_layersÚ_vmap_remove_layersFccsXtrJ‚tdd„|Dƒƒ}t|ƒ}z"t|ƒdadVWdat|ƒndat|ƒ0dS)Ncss$|]}t|tƒs|j|jfVqdS)N)Ú isinstanceÚintÚ_levelÚsize)Ú.0Úd©r úV/var/www/html/assistant/venv/lib/python3.9/site-packages/functorch/dim/batch_tensor.pyÚ óz!_enable_layers..TF)Ú_enabledÚsortedÚlenrr)ÚdimsÚinputÚnr r r Ú_enable_layerss ÿrN)Ú contextlibrZtorch._C._functorchrrrrr r r r Ús