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=<
input-tensor
depth:0		(1, 10, 128)
> fillcolor=lightyellow] 1 [label=<
RNN
depth:1		 input: (1, 10, 128)
output: (1, 10, 128), (1, 1, 128)
> fillcolor=darkseagreen1] 2 [label=<
Dropout
depth:1		 input: (1, 10, 128)
output: (1, 10, 128)
> fillcolor=darkseagreen1] 3 [label=<
__getitem__
depth:1		 input: (1, 10, 128)
output: (1, 128)
> fillcolor=aliceblue] 4 [label=<
Linear
depth:1		 input: (1, 128)
output: (1, 1)
> fillcolor=darkseagreen1] 5 [label=<
output-tensor
depth:0		(1, 1)
> fillcolor=lightyellow] 0 -> 1 1 -> 2 2 -> 3 3 -> 4 4 -> 5 }