doc fix relu --> tanh

Mark Evans · Mark Evans · commit 5fa992e8c1b6 · 2024-11-01T13:37:05.000Z
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
