2017-08-19 19:52:29 +00:00
|
|
|
# Chain
|
|
|
|
|
|
2017-06-05 15:08:23 +00:00
|
|
|
type Chain
|
2016-08-25 21:49:21 +00:00
|
|
|
layers::Vector{Any}
|
2017-03-17 16:34:51 +00:00
|
|
|
Chain(xs...) = new([xs...])
|
2016-08-25 21:49:21 +00:00
|
|
|
end
|
|
|
|
|
|
2017-03-17 16:34:51 +00:00
|
|
|
@forward Chain.layers Base.getindex, Base.first, Base.last, Base.endof, Base.push!
|
2017-03-07 14:37:37 +00:00
|
|
|
@forward Chain.layers Base.start, Base.next, Base.done
|
2016-08-25 21:49:21 +00:00
|
|
|
|
|
|
|
|
(s::Chain)(x) = foldl((x, m) -> m(x), x, s.layers)
|
2017-06-12 11:39:34 +00:00
|
|
|
|
2017-08-19 19:04:21 +00:00
|
|
|
Compiler.graph(s::Chain) =
|
2016-10-25 22:10:35 +00:00
|
|
|
foldl((v, m) -> vertex(m, v), constant(inputnode(1)), s.layers)
|
2016-09-06 17:03:39 +00:00
|
|
|
|
2017-02-28 16:42:48 +00:00
|
|
|
Base.getindex(c::Chain, i::AbstractArray) = Chain(c.layers[i]...)
|
2017-08-19 19:52:29 +00:00
|
|
|
|
|
|
|
|
# Affine
|
|
|
|
|
|
|
|
|
|
struct Affine{S,T}
|
|
|
|
|
W::S
|
|
|
|
|
b::T
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
Affine(in::Integer, out::Integer; init = initn) =
|
|
|
|
|
Affine(track(init(out, in)),
|
|
|
|
|
track(init(out)))
|
|
|
|
|
|
|
|
|
|
(a::Affine)(x) = a.W*x .+ a.b
|
