doc fix relu --> tanh (#601)

Co-authored-by: Mark Evans <[email protected]>
elixir-nx · Nov 2, 2024 · d1dc13b · d1dc13b
1 parent 4a66ba4
commit d1dc13b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/axon.ex b/lib/axon.ex
@@ -3773,7 +3773,7 @@ defmodule Axon do
   layers with `:tanh` layers:
 
       tanh_rewriter = fn [%Axon{} = x], _output ->
-        Axon.relu(x)
+        Axon.tanh(x)
       end
 
       Axon.rewrite_nodes(model, fn