from tensorflow.keras.utils import plot_model plot_model(model, to_file='model.png', show_shapes=True) # 保存模型结构图