pytorchviz
pytorchviz copied to clipboard
Show more custom information of tensor and gradient
- add support to show multiple lines attribute value
- allowed to customize which attr of the variable can be displayed
This is only a demo version, it seems Graphviz always displays the attribute with center alignment since this tool uses caption to contain attributes. Better to display them with left alignment [TODO]