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

todone: nicer syntax on 0.7

Browse Source
This commit is contained in:
Mike J Innes 2018-09-05 15:55:08 +01:00
parent 8b71350878
commit ec16a2c77d
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/layers/recurrent.jl
View File

@ -128,8 +128,7 @@ function LSTMCell(in::Integer, out::Integer;
return cell return cell
end end
function (m::LSTMCell)(h_, x) function (m::LSTMCell)((h, c), x)
h, c = h_ # TODO: nicer syntax on 0.7
b, o = m.b, size(h, 1) b, o = m.b, size(h, 1)
g = m.Wi*x .+ m.Wh*h .+ b g = m.Wi*x .+ m.Wh*h .+ b
input = σ.(gate(g, o, 1)) input = σ.(gate(g, o, 1))