From 496dbfabd2a5725d7737bf4d127cf559541cf6ed Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 16 Jan 2019 14:51:37 +0000 Subject: [PATCH] make chain collectable --- src/layers/basic.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/layers/basic.jl b/src/layers/basic.jl index fddd4fc9..758aa0a9 100644 --- a/src/layers/basic.jl +++ b/src/layers/basic.jl @@ -21,8 +21,8 @@ struct Chain{T<:Tuple} Chain(xs...) = new{typeof(xs)}(xs) end -@forward Chain.layers Base.getindex, Base.first, Base.last, Base.lastindex -@forward Chain.layers Base.iterate +@forward Chain.layers Base.getindex, Base.length, Base.first, Base.last, + Base.iterate, Base.lastindex children(c::Chain) = c.layers mapchildren(f, c::Chain) = Chain(f.(c.layers)...)