You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
from torch import cond # noqa: F401
|
|
from torch._higher_order_ops.map import ( # noqa: F401
|
|
_stack_pytree,
|
|
_unstack_pytree,
|
|
map,
|
|
)
|
