未验证 提交 425dbb28 编写于 作者: H HydrogenSulfate 提交者: GitHub

【cherry-pick】Add jacobian and hessian (#53304)

* cherry pick

* remove 1 unitest of Hessian for numerical result is wrong

* remove 1 unitest of Hessian for numerical result is wrong

* refine docstring; remove incubate usage in autograd.py
上级 8413e4c3
...@@ -18,12 +18,15 @@ from ..fluid.dygraph.base import no_grad_ as no_grad # noqa: F401 ...@@ -18,12 +18,15 @@ from ..fluid.dygraph.base import no_grad_ as no_grad # noqa: F401
from ..fluid.dygraph.base import is_grad_enabled # noqa: F401 from ..fluid.dygraph.base import is_grad_enabled # noqa: F401
from ..fluid.dygraph.base import set_grad_enabled # noqa: F401 from ..fluid.dygraph.base import set_grad_enabled # noqa: F401
from . import backward_mode # noqa: F401 from . import backward_mode # noqa: F401
from .autograd import jacobian, hessian # noqa: F401
from .backward_mode import backward # noqa: F401 from .backward_mode import backward # noqa: F401
from .py_layer import PyLayer # noqa: F401 from .py_layer import PyLayer # noqa: F401
from .py_layer import PyLayerContext # noqa: F401 from .py_layer import PyLayerContext # noqa: F401
from .saved_tensors_hooks import saved_tensors_hooks from .saved_tensors_hooks import saved_tensors_hooks
__all__ = [ # noqa __all__ = [ # noqa
'jacobian',
'hessian',
'backward', 'backward',
'PyLayer', 'PyLayer',
'PyLayerContext', 'PyLayerContext',
......
此差异已折叠。
...@@ -15,6 +15,7 @@ foreach(TEST_OP ${TEST_OPS}) ...@@ -15,6 +15,7 @@ foreach(TEST_OP ${TEST_OPS})
py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS ${GC_ENVS}) py_test_modules(${TEST_OP} MODULES ${TEST_OP} ENVS ${GC_ENVS})
endforeach() endforeach()
set_tests_properties(test_autograd_dynamic PROPERTIES TIMEOUT 100)
set_tests_properties(test_autograd_functional_dynamic PROPERTIES TIMEOUT 200) set_tests_properties(test_autograd_functional_dynamic PROPERTIES TIMEOUT 200)
set_tests_properties(test_autograd_functional_static PROPERTIES TIMEOUT 160) set_tests_properties(test_autograd_functional_static PROPERTIES TIMEOUT 160)
set_tests_properties(test_minimize PROPERTIES TIMEOUT 60) set_tests_properties(test_minimize PROPERTIES TIMEOUT 60)
......
此差异已折叠。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册