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

simplify

Browse Source
This commit is contained in:
Michael Abbott 2019-09-25 13:59:32 +02:00
parent 1a1a96571a
commit 2de84ce79f
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/layers/basic.jl
View File

@ -211,13 +211,9 @@ end
@functor SkipConnection
function (skip::SkipConnection)(input)
#We apply the layers to the input and return the result of the application of the layers and the original input
skip.connection(skip.layers(input), input)
end
function Base.show(io::IO, b::SkipConnection)
print(io, "SkipConnection(")
b.layers isa Chain ? print(io, "Chain(", join(b.layers, ", "), "), ") :
print(io, b.layers, ", ")
print(io, b.connection, ")")
print(io, "SkipConnection(", b.layers, ", ", b.connection, ")")
end