From 06f2ee2284e1432f18783f2f8ea92fd418b8603a Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Tue, 7 Mar 2017 14:37:37 +0000 Subject: [PATCH] iterate over chain --- src/layers/chain.jl | 1 + 1 file changed, 1 insertion(+) diff --git a/src/layers/chain.jl b/src/layers/chain.jl index bb2911ff..43d09e4d 100644 --- 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)