首先安装需求库
pip install pydot-ng
https://graphviz.gitlab.io/download/下载grapgviz指定版本,添加进path
cmd命令行要运行一次dot -c 不然会报错
然后使用tf.keras.utils.plot_model( model, to_file='model.png', show_shapes=False, show_layer_names=True, rankdir='TB', expand_nested=False, dpi=96 )
就可以得到模型图