add Higher derivative vjp (#9)

* add Higher derivative vjp * add autotest code * add autograd.functional.vhp and motified functional * Merge Testcase * Rm chinese chars Co-authored-by: Zhenhua <1209435+hengzi@users.noreply.github.com> Co-authored-by: Huang Zhenhua <huangzhenhua@zhejianglab.com>
ZJLabDubhe · Jun 21, 2022 · a4309ac · a4309ac
1 parent 6fb9880
commit a4309ac
Show file tree

Hide file tree

Showing 4 changed files with 515 additions and 39 deletions.
diff --git a/python/oneflow/autograd/__init__.py b/python/oneflow/autograd/__init__.py
@@ -23,7 +23,7 @@
     is_grad_enabled,
     no_grad,
 )
-from oneflow.autograd.functional import jacobian, hessian
+from oneflow.autograd.functional import vjp, vhp, jacobian, hessian
 
 __all__ = [
     "backward",
@@ -34,6 +34,8 @@
     "inference_mode",
     "is_grad_enabled",
     "no_grad",
+    "vjp",
+    "vhp",
     "jacobian",
     "hessian",
 ]