From 0fe3ac4e770de17a46d37809238a6deae06f98a3 Mon Sep 17 00:00:00 2001
From: dsweber2 <david.weber2@gmail.com>
Date: Tue, 8 Oct 2019 23:05:22 -0700
Subject: [PATCH] bring activations into function call

---
 src/layers/basic.jl | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/layers/basic.jl b/src/layers/basic.jl
index b92bc919..db491424 100644
--- 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)