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

nicer show

Browse Source
This commit is contained in:
Mike J Innes 2017-08-21 17:20:09 +01:00
parent 227e41c37b
commit 3e0f45046c
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/layers/basic.jl
View File

@ -15,6 +15,12 @@ Compiler.graph(s::Chain) =
Base.getindex(c::Chain, i::AbstractArray) = Chain(c.layers[i]...)
function Base.show(io::IO, c::Chain)
print(io, "Chain(")
join(io, c.layers, ", ")
print(io, ")")
end
# Linear
struct Linear{F,S,T}
@ -27,3 +33,9 @@ Linear(in::Integer, out::Integer, σ = identity; init = initn) =
Linear(σ, track(init(out, in)), track(init(out)))
(a::Linear)(x) = a.σ.(a.W*x .+ a.b)
function Base.show(io::IO, l::Linear)
print(io, "Linear(", size(l.W, 2), ", ", size(l.W, 1))
l.σ == identity || print(io, ", ", l.σ)
print(io, ")")
end