Flux.jl/src/layers/affine.jl
2017-03-17 16:34:54 +00:00

14 lines
234 B
Julia

export Affine
@net type Affine
W
b
x -> x*W + b
end
Affine(in::Integer, out::Integer; init = initn) =
Affine(init(in, out), init(1, out))
inferred(::Type{Affine}, in::Tuple{Dims{2}}, out::Integer) =
Affine(in[1][2], out)