是否可以制作张量流图摘要？答案

【问题标题】：Is it possible to make tensorflow graph summary?是否可以制作张量流图摘要？
【发布时间】：2018-07-26 05:14:16
【问题描述】：

我知道 Tensorboard 以及它有多棒，但我认为带有当前图形摘要的简单控制台输出对于原型设计来说更好（并且更快）。并且还知道我可以在与最后一个网络节点运行会话后生成张量板图，如图所示here。

我正在寻找类似于来自 Keras 的 model.summary() 的内容。

换句话说：如何迭代 tensorflow 图并仅打印出自定义高端层，其形状和 dtype 以相同的顺序生成所有这些层？

【问题讨论】：

这能解决您的问题吗？ tensorflow.org/tensorboard/graphs
很容易提取dict格式的摘要stackoverflow.com/a/68128858/10375049
使用可以tf.summary.graph。更多详情请参考tensorflow.org/api_docs/python/tf/summary/graph

【解决方案1】：

这当然是可能的。如果您使用tf.keras 包装器构建您可以轻松地可视化图形，甚至在model.compile() 方法执行之前。它是 keras 内置功能，称为 plot_model()。
*此方法依赖于graphviz 和pydot 库。
对于 pydot 安装：pip install pydot
但是对于graphviz 安装，您必须按照this page 中的步骤操作。而且您可能还必须重新启动机器，因为它会在那里创建系统环境变量。
有关如何使用此方法的教程，请关注this link

【讨论】：

【解决方案2】：

要在训练之前使用形状和数据类型绘制模型，您可以使用：

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

其中“模型”是您构建的模型。模型的输出可能如下所示：

【讨论】：