pytorch打印网络结构的实例

最简单的方法当然可以直接print（net），但是这样网络比较复杂的时候效果不太好，看着比较乱；以前使用caffe的时候有一个网站可以在线生成网络框图，tensorflow可以用tensor board，keras中可以用model.summary()、或者plot_model()。pytorch没有这样的API，但是可以用代码来完成。 （1）安装环境：graphviz conda install -n pytorch python-graphviz 或： sudo apt-get install graphviz 或者从官网下载，按此教程。 （2）生成网络结构的代码： def make