-model = nn.DAG()
-
-model:addEdge(a, b)
-model:addEdge(b, nn.Linear(10, 5), nn.ReLU(), nn.Linear(5, 10), c)
-model:addEdge(b, d)
-model:addEdge(c, e)
-model:addEdge(d, e)
-model:addEdge(d, f)
-model:addEdge(e, g)
-model:addEdge(f, nn.Mul(-1), g)
+model:connect(a, b, c)
+model:connect(b, nn.Linear(10, 15), nn.ReLU(), d)
+model:connect(d, e)
+model:connect(c, d)
+model:connect(c, nn.Mul(-1), e)