mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
913f5784d7
Did a very quick sanity check - it looks like functorch docs don't get the nice preview link that pytofch-bot gives for normal pytorch docs, so I built locally and scanned `html/generated/functorch.functionalize.html` Pull Request resolved: https://github.com/pytorch/pytorch/pull/85742 Approved by: https://github.com/zou3519
32 lines
1012 B
Python
32 lines
1012 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
|
|
from . import _C
|
|
|
|
# Top-level APIs. Please think carefully before adding something to the
|
|
# top-level namespace:
|
|
# - private helper functions should go into functorch._src
|
|
# - very experimental things should go into functorch.experimental
|
|
# - compilation related things should go into functorch.compile
|
|
|
|
# functorch transforms
|
|
from ._src.vmap import vmap
|
|
from ._src.eager_transforms import (
|
|
grad, grad_and_value, vjp, jacrev, jvp, jacfwd, hessian, functionalize
|
|
)
|
|
from ._src.python_key import make_fx
|
|
|
|
# utilities. Maybe these should go in their own namespace in the future?
|
|
from ._src.make_functional import (
|
|
make_functional_with_buffers,
|
|
make_functional,
|
|
combine_state_for_ensemble,
|
|
FunctionalModule,
|
|
FunctionalModuleWithBuffers,
|
|
)
|
|
|
|
__version__ = torch.__version__
|