Fix minor typo in cell (#2692)

* Fix minor typo in cell One of the arguments to `hvp` wasn't being used, which made the example slightly confusing. * Fix both definitions of hvp in the autodiff cookbook. Co-authored-by: Peter Hawkins <phawkins@google.com>
jax-ml · Apr 14, 2020 · d6ab70c · d6ab70c
1 parent 7f6ede1
commit d6ab70c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/notebooks/autodiff_cookbook.ipynb b/docs/notebooks/autodiff_cookbook.ipynb
@@ -462,7 +462,7 @@
    "outputs": [],
    "source": [
     "def hvp(f, x, v):\n",
-    "    return grad(lambda x: np.vdot(grad(f)(x), v))"
+    "    return grad(lambda x: np.vdot(grad(f)(x), v))(x)"
    ]
   },
   {
@@ -899,7 +899,7 @@
    "outputs": [],
    "source": [
     "def hvp(f, x, v):\n",
-    "    return grad(lambda x: np.vdot(grad(f)(x), v))"
+    "    return grad(lambda x: np.vdot(grad(f)(x), v))(x)"
    ]
   },
   {