tensorflow - 如何在张量板上可视化张量摘要

Question

我正在尝试在张量板上可视化张量摘要。但是我在板上根本看不到张量摘要。这是我的代码:

        out = tf.strided_slice(logits, begin=[self.args.uttWindowSize-1, 0], end=[-self.args.uttWindowSize+1, self.args.numClasses],
                               strides=[1, 1], name='softmax_truncated')
        tf.summary.tensor_summary('softmax_input', out)

其中 out 是多维张量。我想我的代码一定有问题。可能我错误地使用了 tensor_summary 函数。

Answer 1

您所做的是创建一个摘要操作，但不调用它也不编写摘要(请参阅 documentation )。要实际创建摘要，您需要执行以下操作:

# Create a summary operation
summary_op = tf.summary.tensor_summary('softmax_input', out)

# Create the summary
summary_str = sess.run(summary_op)

# Create a summary writer
writer = tf.train.SummaryWriter(...)

# Write the summary
writer.add_summary(summary_str)

仅当您没有像 Supervisor 这样的更高级别的帮助程序时，才需要明确编写摘要(最后两行)。 。否则你调用

sv.summary_computed(sess, summary_str)

主管会处理它。

更多信息，另请参阅: How to manually create a tf.Summary()