2016-11-14 22:16:00 +00:00
|
|
|
export Affine
|
2016-08-22 20:13:28 +00:00
|
|
|
|
2016-11-14 22:16:00 +00:00
|
|
|
@net type Affine
|
2016-08-22 13:49:41 +00:00
|
|
|
W
|
|
|
|
|
b
|
2016-12-15 22:57:36 +00:00
|
|
|
x -> x*W + b
|
2016-08-22 13:49:41 +00:00
|
|
|
end
|
|
|
|
|
|
2016-11-14 22:16:00 +00:00
|
|
|
Affine(in::Integer, out::Integer; init = initn) =
|
|
|
|
|
Affine(init(in, out), init(1, out))
|
2017-03-17 16:34:51 +00:00
|
|
|
|
|
|
|
|
inferred(::Type{Affine}, in::Tuple{Dims{2}}, out::Integer) =
|
|
|
|
|
Affine(in[1][2], out)
|
