1from torch import cond # noqa: F401 2from torch._higher_order_ops.cond import UnsupportedAliasMutationException # noqa: F401 3from torch._higher_order_ops.map import ( # noqa: F401 4 _stack_pytree, 5 _unstack_pytree, 6 map, 7) 8
1from torch import cond # noqa: F401 2from torch._higher_order_ops.cond import UnsupportedAliasMutationException # noqa: F401 3from torch._higher_order_ops.map import ( # noqa: F401 4 _stack_pytree, 5 _unstack_pytree, 6 map, 7) 8