| 12345678910111213141516 |
- # This file has moved to under torch/_functorch. It is not public API.
- # If you are not a PyTorch developer and you are relying on the following
- # imports, please file an issue.
- from torch._functorch.vmap import (
- _add_batch_dim,
- _broadcast_to_and_flatten,
- _get_name,
- _remove_batch_dim,
- _validate_and_get_batch_size,
- Tensor,
- tree_flatten,
- tree_unflatten,
- _process_batched_inputs,
- _create_batched_inputs,
- _unwrap_batched,
- )
|
