diff --git a/src/layers/basic.jl b/src/layers/basic.jl index 59349aba..a6a8bd62 100644 --- a/src/layers/basic.jl +++ b/src/layers/basic.jl @@ -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