mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
4068c5467d
This will be the last disruptive functorch internals change. Why are we moving these files? - As a part of rationalizing functorch we are moving the code in functorch/_src to torch/_functorch - This is so that we can offer the functorch APIs as native PyTorch APIs (coming soon) and resolve some internal build issues. Why are we moving all of these files at once? - It's better to break developers all at once rather than many times Test Plan: - wait for tests Pull Request resolved: https://github.com/pytorch/pytorch/pull/90091 Approved by: https://github.com/anijain2305, https://github.com/ezyang
32 lines
1.0 KiB
Python
32 lines
1.0 KiB
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
|
|
from . import _C
|
|
|
|
# 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
|
|
|
|
# functorch transforms
|
|
from torch._functorch.vmap import vmap
|
|
from torch._functorch.eager_transforms import (
|
|
grad, grad_and_value, vjp, jacrev, jvp, jacfwd, hessian, functionalize
|
|
)
|
|
from torch._functorch.python_key import make_fx
|
|
|
|
# utilities. Maybe these should go in their own namespace in the future?
|
|
from torch._functorch.make_functional import (
|
|
make_functional_with_buffers,
|
|
make_functional,
|
|
combine_state_for_ensemble,
|
|
FunctionalModule,
|
|
FunctionalModuleWithBuffers,
|
|
)
|
|
|
|
__version__ = torch.__version__
|