added return_state to cornn

MartinuzziFrancesco · Feb 9, 2025 · 1526cc5 · 1526cc5
1 parent 3a8718b
commit 1526cc5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/cells/cornn_cell.jl b/src/cells/cornn_cell.jl
@@ -109,7 +109,9 @@ See [`coRNNCell`](@ref) for a layer that processes a single sequence.
 - `init_kernel`: initializer for the input to hidden weights
 - `init_recurrent_kernel`: initializer for the hidden to hidden weights
 - `bias`: include a bias or not. Default is `true`
-
+- `return_state`: Option to return the last state together with the output.
+  Default is `false`.
+  
 # Equations
 ```math
 \begin{aligned}