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

make activations zygote friendly

Browse Source
This commit is contained in:
dsweber2 2019-09-10 00:54:49 -07:00
parent 9d05afaccc
commit cdaaca8cfa
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/layers/basic.jl
View File

@ -44,17 +44,15 @@ end
# it might be replaced in the future for better performance # it might be replaced in the future for better performance
# see issue https://github.com/FluxML/Flux.jl/issues/702 # see issue https://github.com/FluxML/Flux.jl/issues/702
# Johnny Chen -- @johnnychen94 # Johnny Chen -- @johnnychen94
# only slightly changed to better handle interaction with Zygote @dsweber2
""" """
activations(c::Chain, input) activations(c::Chain, input)
Calculate the forward results of each layers in Chain `c` with `input` as model input. Calculate the forward results of each layers in Chain `c` with `input` as model input.
""" """
function activations(c::Chain, input) function activations(c::Chain, input)
rst = [] buffed = accumulate!((x,y)->y(x), Zygote.Buffer([], length(c)),
for l in c [l for l in c], dims=1, init=input)
x = get(rst, length(rst), input) return copy(buffed)
push!(rst, l(x))
end
return rst
end end