使用 TensorBoard 查看 TensorFlow 图形并设置 tag

TensorBoard 是一个强大的工具，可以用来可视化 TensorFlow 模型的训练过程和图形结构。在使用 TensorBoard 查看图形时，我们可以通过设置 tag 来区分不同版本的模型或不同阶段的图形。

如何设置 tag

要设置 tag，可以使用 tf.summary.FileWriter 的 add_graph 方法。

with tf.compat.v1.Session(graph=graph) as sess:
    # 创建摘要写入器
    writer = tf.compat.v1.summary.FileWriter('logs', graph=sess.graph)
    # 设置tag
    writer.add_graph(sess.graph, tag='my_graph')
    writer.close()

在这个例子中，使用 add_graph 方法来添加图形，并为它设置了一个 tag 为 'my_graph'。这样就可以在 TensorBoard 中以这个 tag 来查看图形了。

使用 tag 查看图形

在 TensorBoard 中，可以使用 Graphs 面板查看所有已添加的图形。可以通过选择不同的 tag 来切换查看不同的图形。

总结

通过设置 tag，我们可以更好地组织和管理 TensorFlow 图形，方便在 TensorBoard 中查看和比较不同版本的模型。