How could i properly visualize LSTM in pytorch?

i know this link this
but when i feed it the output of the RNN or LSTM it gives a complecated graph.
is there a way to avoid that?

No way yet, but better visualization will be possible in the future after the autograd refactor is merged

1 Like