X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=README.md;h=b122ee02b215bbac8facd672a361a6ef9cf29a1f;hb=d8fd868f94ce0b66cd2cc1a4615df10a88b5d5ec;hp=2ff5b9966b69eb4031506fd123b2fa2fc32bd961;hpb=9dad4fa1118632bfa02c01e4d6a8a5a129061a54;p=dagnn.git diff --git a/README.md b/README.md index 2ff5b99..b122ee0 100644 --- a/README.md +++ b/README.md @@ -1,7 +1,9 @@ +#Introduction# + This package implements a new module nn.DAG which inherits from nn.Container and allows to combine modules in an arbitrary graph without cycle. -#Example# +##Example## A typical use would be: @@ -42,12 +44,54 @@ and run a forward pass with a random batch of 30 samples. Note that DAG:connect allows to add a bunch of edges at once. This is particularly useful to add anonymous modules which have a single predecessor and successor. -#Input and output# +##Input and output## -If a node has a single successor, its output is sent unchanged as input to that successor. If it has multiple successors, the outputs are collected into a table, and the table is used as input to the successor node. The indexes of the outputs in that table reflects the order of the DAG:connect() commands. +If a node has a single predecessor, its output is taken as-is. If it has multiple predecessors, all the outputs are collected into a table, and the table is used as input. The indexes of the outputs in that table reflects the order in which the predecessors appeared in the DAG:connect() commands. -The expected input (respectively the produced output) is a nested table of inputs reflecting the structure of the nested table of modules provided to DAG:setInput (respectively DAG:setOutput) +The input to the DAG (respectively the produced output) is a nested table of inputs reflecting the structure of the nested table of modules provided to DAG:setInput (respectively DAG:setOutput) So for instance, in the example above, the model expects a tensor as input, since it is the input to the module a, and its output will is a table composed of two tensors, corresponding to the outputs of d and e respectively. +#Usage# + +##nn.DAG()## + +Create a new empty DAG, which inherits from nn.Container. + +##nn.DAG:connect([module1 [, module2 [, ...]]])## + +Add new nodes corresponding to the modules passed as arguments if they +are not already existing. Add edges between every the nodes +corresponding to pairs of successive modules. + +##nn.DAG:setInput(i)## + +Defines the content and structure of the input. The argument should be +either a module, or a (nested) table of module. The input to the DAG +should be a (nested) table of inputs with the corresponding structure. + +##nn.DAG:setOutput(o)## + +Same as DAG:setInput. + +##nn.DAG:print()## + +Prints the list of nodes. + +##nn.DAG:saveDot(filename)## + +Save a dot file to be used by the Graphviz set of tools for graph visualization. + +##nn.DAG:updateOutput(input)## + +See the torch documentation. + +##nn.DAG:updateGradInput(input, gradOutput)## + +See the torch documentation. + +##nn.DAG:accGradParameters(input, gradOutput, scale)## + +See the torch documentation. + *Francois Fleuret, Jan 13th, 2017*