Flux.jl/src/layers/dense.jl

11 lines
143 B
Julia
Raw Normal View History

2016-08-22 20:13:28 +00:00
export Dense
2016-08-22 13:49:41 +00:00
@model type Dense
W
b
x -> W*x + b
end
Dense(in::Integer, out::Integer; init = randn) =
Dense(init(out, in), init(out))