typing

MartinuzziFrancesco · Dec 16, 2024 · 3c4eff5 · 3c4eff5
1 parent 487426c
commit 3c4eff5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/RecurrentLayers.jl b/src/RecurrentLayers.jl
@@ -35,7 +35,7 @@ function (rlayer::AbstractRecurrentLayer)(inp::AbstractVecOrMat)
     return rlayer(inp, state)
 end
 
-function (rlayer::AbstractRecurrentLayer)(inp, state)
+function (rlayer::AbstractRecurrentLayer)(inp::AbstractArray, state::AbstractVecOrMat)
     @assert ndims(inp) == 2 || ndims(inp) == 3
     return scan(rlayer.cell, inp, state)
 end
@@ -47,7 +47,6 @@ export MGU, LiGRU, IndRNN, RAN, LightRU, NAS, RHN, MUT1, MUT2, MUT3,
 SCRN, PeepholeLSTM, FastRNN, FastGRNN
 
 
-#TODO add double bias
 include("mgu_cell.jl")
 include("ligru_cell.jl")
 include("indrnn_cell.jl")