tensorflow tf.keras.utils.plot

tensorflow tf.keras.utils.plot 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 # 图片每英寸点数。
)