netron软件;微软的tensorwatch ;tensorboardX ;或者graphviz加上torchviz等都可以实现神经网络的可视化。 本人一直使用的tensorboardX,针对其他的使用不多,netron对pytorch框架的效果不好。