diff --git a/src/layers/dense.jl b/src/layers/dense.jl index e69de29b..a36cf1eb 100644 --- a/src/layers/dense.jl +++ b/src/layers/dense.jl @@ -0,0 +1,8 @@ +@model type Dense + W + b + x -> W*x + b +end + +Dense(in::Integer, out::Integer; init = randn) = + Dense(init(out, in), init(out))