tensorflow tf.keras.utils.plot_model 画网络拓扑图
# pip install graphviz
# pip install pydot
# 下载 graphviz，并添加到环境变量。 https://graphviz.gitlab.io/download/
import tensorflow as tf
import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files/Graphviz/bin/' # graphviz的安装路径
tf.keras.utils.plot_model(
model, # keras模型
to_file="tensorflow 多输入 多输出 模型1.png", # 保存图片路径
show_shapes=True, # 是否显示形状信息
show_layer_names=True, # 是否显示图层名称
rankdir="TB", # "TB":垂直图 "LR":水平图
expand_nested=True, # 是否将嵌套模型展开为簇。
dpi=96 # 图片每英寸点数。
)
