From 7d782000870bf8e7d20a45b07022288be5befcb5 Mon Sep 17 00:00:00 2001 From: Aurora Rossi <65721467+aurorarossi@users.noreply.github.com> Date: Mon, 19 Aug 2024 15:29:05 +0200 Subject: [PATCH] Fix Co-authored-by: Carlo Lucibello --- GNNLux/src/layers/temporalconv.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/GNNLux/src/layers/temporalconv.jl b/GNNLux/src/layers/temporalconv.jl index f9180dc11..2ab6235b3 100644 --- a/GNNLux/src/layers/temporalconv.jl +++ b/GNNLux/src/layers/temporalconv.jl @@ -12,8 +12,8 @@ function (r::StatefulRecurrentCell)(g, x::AbstractMatrix, ps, st::NamedTuple) end function (r::StatefulRecurrentCell)(g, x::AbstractVector, ps, st::NamedTuple) - (out, carry), st = applyrecurrentcell(r.cell, g, first(x), ps, st.cell, st.carry) - for xᵢ in x[(begin + 1):end] + st, carry = st.cell, st.carry + for xᵢ in x (out, carry), st = applyrecurrentcell(r.cell, g, xᵢ, ps, st, carry) end return out, (; cell=st, carry)