fix lstm

2016-12-15 20:53:08 +00:00 · 2016-12-15 20:53:08 +00:00 · 4b64bf11a5
parent c6fb9c1f0c
commit 4b64bf11a5
1 changed files with 1 additions and 1 deletions
--- a/src/layers/recurrent.jl
+++ b/src/layers/recurrent.jl
@ -48,4 +48,4 @@ end

 LSTM(in, out; init = initn) =
  LSTM(vcat([[init((in, out)), init((out, out)), init((1, out))] for _ = 1:4]...)...,
-       zeros(Float32, 1, out), zeros(Float32, 1, out))
+       zeros(Float32, out), zeros(Float32, out))