Fixed Axon.gru return value in docs

elixir-nx · Feb 6, 2025 · 11b0332 · 11b0332
1 parent 953f34d
commit 11b0332
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/axon.ex b/lib/axon.ex
@@ -2801,7 +2801,7 @@ defmodule Axon do
   GRUs apply `Axon.Layers.gru_cell/7` over an entire input
   sequence and return:
 
-      {{new_hidden}, output_sequence}
+      {output_sequence, {new_hidden}}
 
   You can use the output state as the hidden state of another
   GRU layer.