在 tensorboard 中可视化边缘厚度

Question

我希望图形可视化工具用张量维度和边缘厚度来标记边，以反映总张量大小。基本和this doc:

里面写的一模一样

When the serialized GraphDef includes tensor shapes, the graph visualizer labels edges with tensor dimensions, and edge thickness reflects total tensor size. To include tensor shapes in the GraphDef pass the actual graph object (as in sess.graph) to the FileWriter when serializing the graph. The images below show the CIFAR-10 model with tensor shape information:

我将图形对象传递给我 summary.FileWriter:

writer = tf.summary.FileWriter(_dir_tensorboard, graph=sess.graph, flush_secs=300)

但是我没有得到任何关于 thickness 的信息（所有线的大小都相同）。我只有关于张量形状的信息和关于张量数量的信息。

如何才能达到教程声称的相同视觉效果？

Answer 1

有一个影响 1.11 和 1.12 的回归。它应该用 https://github.com/tensorflow/tensorboard/pull/1544 修复。很抱歉 :( 下次发生这种情况时请提交 GitHub 问题。