I’m wondering if anyone has created a tool that will create some kind of dataflow graph of a PyTorch model showing the sizes of data at various points in the model. Even something that generates graphviz would be helpful.