Skip to content

Commit

Permalink
small cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
MartinuzziFrancesco committed Dec 15, 2024
1 parent bebd3b2 commit f9596d9
Showing 1 changed file with 0 additions and 6 deletions.
6 changes: 0 additions & 6 deletions src/scrn_cell.jl
Original file line number Diff line number Diff line change
Expand Up @@ -68,12 +68,6 @@ function SCRNCell((input_size, hidden_size)::Pair;
return SCRNCell(Wi, Wh, Wc, b, alpha)
end

function (scrn::SCRNCell)(inp::AbstractVecOrMat)
state = zeros_like(inp, size(scrn.Wh, 2))
c_state = zeros_like(state)
return scrn(inp, (state, c_state))
end

function (scrn::SCRNCell)(inp::AbstractVecOrMat, (state, c_state))
_size_check(scrn, inp, 1 => size(scrn.Wi,2))
Wi, Wh, Wc, b = scrn.Wi, scrn.Wh, scrn.Wc, scrn.bias
Expand Down

0 comments on commit f9596d9

Please sign in to comment.