pytorch

1
from torch import cond  # noqa: F401
2
from torch._higher_order_ops.cond import UnsupportedAliasMutationException  # noqa: F401
3

4
from torch._higher_order_ops.map import (  # noqa: F401
5
    _stack_pytree,
6
    _unstack_pytree,
7
    map,
8
)
9