bring activations into function call

2019-10-08 23:05:22 -07:00 · 2019-10-08 23:05:22 -07:00 · 0fe3ac4e77
commit 0fe3ac4e77
parent db92b0e3ce
1 changed files with 2 additions and 0 deletions
--- a/src/layers/basic.jl
+++ b/src/layers/basic.jl
@ -31,6 +31,8 @@ applychain(fs::Tuple, x) = applychain(tail(fs), first(fs)(x))

 (c::Chain)(x) = applychain(c.layers, x)

+(c::Chain)(x, i) = extraChain(c.layers, x)[i]
+
 Base.getindex(c::Chain, i::AbstractArray) = Chain(c.layers[i]...)

 function Base.show(io::IO, c::Chain)