Flux.jl/src/layers/recurrent.jl

13 lines
226 B
Julia
Raw Normal View History

2016-10-30 11:41:52 +00:00
export Recurrent
2016-10-28 23:13:32 +00:00
@net type Recurrent
2016-10-30 16:07:18 +00:00
Wxy; Wyy; by
y
2016-10-28 23:13:32 +00:00
function (x)
2016-10-30 16:07:18 +00:00
y = tanh( x * Wxy + y * Wyy + by )
2016-10-28 23:13:32 +00:00
end
end
2016-10-30 16:07:18 +00:00
Recurrent(in, out; init = initn) =
Recurrent(init((in, out)), init((out, out)), init(out), init(out))