我试图用keras绘制我的模型，如下所示：
# Plot model graph
tf.keras.utils.plot_model(model, to_file='Model1.png')
from IPython.display import Image
Image(retina=True, filename='Model1.png')我得到以下结果：我的模型
但是，我在互联网上的某个地方看到，有人绘制了他的模型，像这样：模型我需要
我怎样才能把代码改成那样？我的模型每一层的输入/输出信息？
1条答案
按热度按时间jchrr9hc1#
您可以使用该参数
show_shapes=True.从
tf.keras文档：显示形状：是否显示形状信息。
（请在此处仔细查看：https://www.tensorflow.org/api_docs/python/tf/keras/utils/plot_model )