mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
b2d703e2d7
Just a small optimization. This PR changes it so that import of functorch.dim ends up loading functorch._C (which is entirely composed of torchdim APIs) Test Plan: - existing tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/100491 Approved by: https://github.com/Chillee, https://github.com/kshitij12345
29 lines
996 B
Python
29 lines
996 B
Python
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
import torch
|
|
|
|
# Top-level APIs. Please think carefully before adding something to the
|
|
# top-level namespace:
|
|
# - private helper functions should go into torch._functorch
|
|
# - very experimental things should go into functorch.experimental
|
|
# - compilation related things should go into functorch.compile
|
|
|
|
# Was never documented
|
|
from torch._functorch.python_key import make_fx
|
|
|
|
from torch._functorch.deprecated import (
|
|
vmap, grad, grad_and_value, vjp, jvp, jacrev, jacfwd, hessian, functionalize,
|
|
make_functional, make_functional_with_buffers, combine_state_for_ensemble,
|
|
)
|
|
|
|
# utilities. Maybe these should go in their own namespace in the future?
|
|
from torch._functorch.make_functional import (
|
|
FunctionalModule,
|
|
FunctionalModuleWithBuffers,
|
|
)
|
|
|
|
__version__ = torch.__version__
|