From 1526cc57126d2aa1ba0887bf8d4550ce40bf099f Mon Sep 17 00:00:00 2001 From: MartinuzziFrancesco Date: Sun, 9 Feb 2025 15:04:45 +0100 Subject: [PATCH] added return_state to cornn --- src/cells/cornn_cell.jl | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/cells/cornn_cell.jl b/src/cells/cornn_cell.jl index 5c1db89..7b44f72 100644 --- 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}