szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs
MIT License
3.24k stars 279 forks source link

add vizualization for recurrent nets #79

Open kapoorlab opened 1 year ago

kapoorlab commented 1 year ago

Recurrent nets have list of tensors that need to be added to the visualization than having just a single tensor which the visualization code assumes. This PR adds we created an isinstance block for the cases where list of tensors is passed on to add_base_tensor function while leaving the original code in the else block.