是否有可能使tensorflow图形摘要？

Question

我知道Tensorboard和它是多么棒，但我认为简单的控制台输出与当前的图形摘要更好(更快)的原型目的。还知道我可以在简单地运行与最后一个网络节点的会话后生成张量板图，如这里所示。

我要找的是类似于Keras的model.summary()。

换句话说:如何在tensorflow图上迭代，只打印出具有相同形状和dtype的自定义高端层，这些层是如何生成的？

Answer 1

这当然是可能的。如果使用tf.keras包装器进行构建，甚至在model.compile()方法执行之前，您就可以轻松地可视化图形。这是名为keras的内置功能，称为plot_model()。

*该方法依赖于graphviz和pydot库。

用于pydot安装：pip install pydot

但是对于graphviz安装，您可以按照此页中的步骤执行。而且，您可能还必须重新启动机器，因为它在那里创建了系统环境变量。

有关如何使用此方法的教程，请遵循此链接

Answer 2

若要在培训前用形状和dtype绘制模型，可以使用：

tf.keras.utils.plot_model(model, show_shapes=True, expand_nested=True, show_dtype=True)

“模型”是你建造的模型。模型的输出可能如下所示：

​