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

iterate over chain

Browse Source
This commit is contained in:
Mike J Innes 2017-03-07 14:37:37 +00:00
parent 2a57150bce
commit 06f2ee2284
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/layers/chain.jl
View File

@ -21,6 +21,7 @@ type Chain <: Model
end end
@forward Chain.layers Base.getindex, Base.first, Base.last, Base.endof @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) (s::Chain)(x) = foldl((x, m) -> m(x), x, s.layers)
back!(s::Chain, Δ) = foldr((m, Δ) -> back!(m, Δ), Δ, s.layers) back!(s::Chain, Δ) = foldr((m, Δ) -> back!(m, Δ), Δ, s.layers)