docs: fixes invalid syntax in Nx.Defn.grad/2 docs (#1369)

elixir-nx · Nov 14, 2023 · 9c97dbe · 9c97dbe
1 parent 49d2467
commit 9c97dbe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nx/lib/nx/defn.ex b/nx/lib/nx/defn.ex
@@ -602,7 +602,7 @@ defmodule Nx.Defn do
   ## Examples
 
       defn tanh_grad(t) do
-        grad(t, &Nx.tanh/&1)
+        grad(t, &Nx.tanh/1)
       end
 
   To differentiate on multiple vars, pass a tuple as first argument:
@@ -658,7 +658,7 @@ defmodule Nx.Defn do
   ## Examples
 
       defn tanh_grad(t) do
-        value_and_grad(t, &Nx.tanh/&1)
+        value_and_grad(t, &Nx.tanh/1)
       end
 
   To differentiate on multiple vars, pass a tuple as first argument: