-- The main use should be to add an edge between two modules, but it
-- can also add a full sequence of modules
-function DAG:addEdge(...)
+function DAG:connect(...)
self.sorted = nil
local prev
for _, nnm in pairs({...}) do
end
----------------------------------------------------------------------
+
+function DAG:dot(filename)
+ local file = (filename and io.open(filename, 'w')) or io.stdout
+
+ file:write('digraph {\n')
+
+ file:write('\n')
+
+ for nnma, node in pairs(self.node) do
+ file:write(
+ ' '
+ .. node.index
+ .. ' [shape=box,label=\"' .. torch.type(nnma) .. '\"]'
+ .. '\n'
+ )
+
+ for _, nnmb in pairs(node.succ) do
+ file:write(
+ ' '
+ .. node.index
+ .. ' -> '
+ .. self.node[nnmb].index
+ .. '\n'
+ )
+ end
+
+ file:write('\n')
+ end
+
+ file:write('}\n')
+
+end