fix recurrent layer batching

This commit is contained in:
Mike J Innes 2017-03-20 23:11:08 +00:00
parent 6237aa6739
commit eadf96605c

View File

@ -9,7 +9,7 @@ export Recurrent, GatedRecurrent, LSTM
end
Recurrent(in, out; init = initn) =
Recurrent(init((in, out)), init((out, out)), init(out), init(out))
Recurrent(init((in, out)), init((out, out)), init(1, out), init(1, out))
@net type GatedRecurrent
Wxr; Wyr; br