iterate over chain

2017-03-07 14:37:37 +00:00 · 2017-03-07 14:37:37 +00:00 · 06f2ee2284
commit 06f2ee2284
parent 2a57150bce
1 changed files with 1 additions and 0 deletions
--- a/src/layers/chain.jl
+++ b/src/layers/chain.jl
@ -21,6 +21,7 @@ type Chain <: Model
 end

@forward Chain.layers Base.getindex, Base.first, Base.last, Base.endof
+@forward Chain.layers Base.start, Base.next, Base.done

 (s::Chain)(x) = foldl((x, m) -> m(x), x, s.layers)
 back!(s::Chain, Δ) = foldr((m, Δ) -> back!(m, Δ), Δ, s.layers)