python - tf.print 不会在正在评估的节点上打印 sess

Question

使用 tf print documentation

我写的

    print_op = tf.print("tensors:", cut_points[0,0,:], output_stream=sys.stderr)
    with tf.control_dependencies([print_op]):
        return cut_points

但不输出到std(我看到其他日志，并且 session 确实评估了这一点。

Answer 1

tf.control_dependencies只影响在上下文中创建的新操作。在您的代码片段中，您没有在上下文中创建任何新操作，因此它没有效果。最简单的解决方案是使用 tf.identity将产生相同结果但具有控制依赖性的操作:

print_op = tf.print("tensors:", cut_points[0,0,:], output_stream=sys.stderr)
with tf.control_dependencies([print_op]):
    return tf.identity(cut_points)