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

bring activations into function call

Browse Source
This commit is contained in:
dsweber2 2019-10-08 23:05:22 -07:00
parent db92b0e3ce
commit 0fe3ac4e77
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/basic.jl
View File

@ -31,6 +31,8 @@ applychain(fs::Tuple, x) = applychain(tail(fs), first(fs)(x))
(c::Chain)(x) = applychain(c.layers, 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]...) Base.getindex(c::Chain, i::AbstractArray) = Chain(c.layers[i]...)
function Base.show(io::IO, c::Chain) function Base.show(io::IO, c::Chain)