python - 如何合并不是 tensorflow 中的所有摘要？

Question

我有一个包含两部分的大图，我依次运行。两者都有摘要。

我正在用节点调用摘要

merged_summary = tf.summary.merge_all()

但注意到，它会导致图的后半部分张量在它有意义之前就被评估。

那么，如何只合并图表一半的摘要？

Answer 1

您可以使用 tf.summary.merge ，传递要合并的摘要列表。例如，如果您有摘要:

cost_summary = tf.summary.scalar('cost_sum', cost)  # for some 'cost' tensor
grad_summary = tf.summary.scalar('grad_sum', grad)  # for some 'grad' tensor

您可以按名称合并它们:

merged = tf.summary.merge([cost_summary, grad_summary])

因此，只需为图表的每个部分创建合并汇总运算符，并在需要时调用它们。