TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix lstm

Browse Source
This commit is contained in:
Mike J Innes 2016-12-15 20:53:08 +00:00
parent c6fb9c1f0c
commit 4b64bf11a5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/recurrent.jl
View File

@ -48,4 +48,4 @@ end
LSTM(in, out; init = initn) = LSTM(in, out; init = initn) =
LSTM(vcat([[init((in, out)), init((out, out)), init((1, out))] for _ = 1:4]...)..., 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))