MLP_Annotated.dot

digraph G {
labelloc="t";
label="MLP";
node [shape = oval]; 
n1[label="output (1)\ninput = {Tensor[10]}\lmodule = nn.Linear(10 -> 1)\lgradOutput = {Tensor[1]}" tooltip="[[string \"local function get_network()...\"]]:9"];
n2[label="Node2\ninput = {Tensor[1]}\lgradOutput = {Tensor[1]}" tooltip="[[C]]:-1"];
n3[label="h2 (3)\ninput = {Tensor[10]}\lmodule = nn.Sigmoid\lgradOutput = {Tensor[10]}" tooltip="[[string \"local function get_network()...\"]]:8"];
n4[label="h1 (4)\ninput = {Tensor[20]}\lmodule = nn.Linear(20 -> 10)\lgradOutput = {Tensor[10]}" tooltip="[[string \"local function get_network()...\"]]:7"];
n5[label="input (5)\ninput = {Tensor[20]}\lmodule = nn.Identity\lgradOutput = {Tensor[20]}" tooltip="[[string \"local function get_network()...\"]]:3"];
n6[label="Node6\ninput = {Tensor[20]}\lgradOutput = {Tensor[20]}" tooltip="[[C]]:-1"];
n1 -> n2;
n3 -> n1;
n4 -> n3;
n5 -> n4;
n6 -> n5;
}