net shim

src/Flux.jl

@@ -19,6 +19,9 @@ using .Tracker
 include("optimise/Optimise.jl")
 using .Optimise
 
+include("graph/Graph.jl")
+using .Graph
+
 include("utils.jl")
 include("onehot.jl")
 include("tree.jl")

src/graph/Graph.jl

@@ -0,0 +1,9 @@
+module Graph
+
+export @net
+
+macro net(ex)
+  esc(ex)
+end
+
+end

src/layers/basic.jl

@@ -56,7 +56,7 @@ Dense(in::Integer, out::Integer, σ = identity; init = initn) =
 
 treelike(Dense)
 
-function (a::Dense)(x)
+@net function (a::Dense)(x)
   W, b, σ = a.W, a.b, a.σ
   σ.(W*x .+ b)
 end