From 5fa992e8c1b6f70212bac81f5d56842d6b584d91 Mon Sep 17 00:00:00 2001
From: Mark Evans <mark@hungryegg.co.uk>
Date: Fri, 1 Nov 2024 13:37:05 +0000
Subject: [PATCH] doc fix `relu` --> `tanh`

---
 lib/axon.ex | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/axon.ex b/lib/axon.ex
index 52f4ba6d..b342c6c3 100644
--- 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