Liquid-Dispensing / rnn_torchview
sw882882's picture
model v1
8a3b924 verified
strict digraph model {
graph [ordering=in rankdir=TB size="12.0,12.0"]
node [align=left fontname="Linux libertine" fontsize=10 height=0.2 margin=0 ranksep=0.1 shape=plaintext style=filled]
edge [fontsize=10]
0 [label=<
<TABLE BORDER="0" CELLBORDER="1"
CELLSPACING="0" CELLPADDING="4">
<TR><TD>input-tensor<BR/>depth:0</TD><TD>(1, 10, 128)</TD></TR>
</TABLE>> fillcolor=lightyellow]
1 [label=<
<TABLE BORDER="0" CELLBORDER="1"
CELLSPACING="0" CELLPADDING="4">
<TR>
<TD ROWSPAN="2">RNN<BR/>depth:1</TD>
<TD COLSPAN="2">input:</TD>
<TD COLSPAN="2">(1, 10, 128) </TD>
</TR>
<TR>
<TD COLSPAN="2">output: </TD>
<TD COLSPAN="2">(1, 10, 128), (1, 1, 128) </TD>
</TR>
</TABLE>> fillcolor=darkseagreen1]
2 [label=<
<TABLE BORDER="0" CELLBORDER="1"
CELLSPACING="0" CELLPADDING="4">
<TR>
<TD ROWSPAN="2">Dropout<BR/>depth:1</TD>
<TD COLSPAN="2">input:</TD>
<TD COLSPAN="2">(1, 10, 128) </TD>
</TR>
<TR>
<TD COLSPAN="2">output: </TD>
<TD COLSPAN="2">(1, 10, 128) </TD>
</TR>
</TABLE>> fillcolor=darkseagreen1]
3 [label=<
<TABLE BORDER="0" CELLBORDER="1"
CELLSPACING="0" CELLPADDING="4">
<TR>
<TD ROWSPAN="2">__getitem__<BR/>depth:1</TD>
<TD COLSPAN="2">input:</TD>
<TD COLSPAN="2">(1, 10, 128) </TD>
</TR>
<TR>
<TD COLSPAN="2">output: </TD>
<TD COLSPAN="2">(1, 128) </TD>
</TR>
</TABLE>> fillcolor=aliceblue]
4 [label=<
<TABLE BORDER="0" CELLBORDER="1"
CELLSPACING="0" CELLPADDING="4">
<TR>
<TD ROWSPAN="2">Linear<BR/>depth:1</TD>
<TD COLSPAN="2">input:</TD>
<TD COLSPAN="2">(1, 128) </TD>
</TR>
<TR>
<TD COLSPAN="2">output: </TD>
<TD COLSPAN="2">(1, 1) </TD>
</TR>
</TABLE>> fillcolor=darkseagreen1]
5 [label=<
<TABLE BORDER="0" CELLBORDER="1"
CELLSPACING="0" CELLPADDING="4">
<TR><TD>output-tensor<BR/>depth:0</TD><TD>(1, 1)</TD></TR>
</TABLE>> fillcolor=lightyellow]
0 -> 1
1 -> 2
2 -> 3
3 -> 4
4 -> 5
}