Flux.jl/test/basic.jl

11 lines
196 B
Julia
Raw Normal View History

2017-01-24 11:53:42 +00:00
xs = randn(10)
2016-12-15 22:31:39 +00:00
d = Affine(10, 20)
2017-01-30 17:08:45 +00:00
@test d(xs) (xs'*d.W.x + d.b.x)[1,:]
2016-12-15 23:11:35 +00:00
2016-12-15 23:13:20 +00:00
let
2016-12-26 12:19:04 +00:00
@capture(syntax(d), _Frame(_Line(x_[1] * W_ + b_)))
2016-12-15 23:13:20 +00:00
@test isa(x, Input) && isa(W, Param) && isa(b, Param)
2016-12-15 23:11:35 +00:00
end