python - Tensorflow 函数打印

Question

我在文件 neural_network.py 中有一个定义损失函数的函数:

def loss(a, b):
    ...
    debug = tf.Print(a, [a], message = 'debug: ')
    debug.eval(session = ???)
    return tf.add(a, b)

解释一下，我想在这个函数的某处打印一个张量。但是，我没有在此函数中声明任何 session ；我的 session 在另一个名为 forecaster.py 的文件中声明。因此，当我尝试将 tf.Print() 放入 loss() 时，我做不到，因为我不知道要评估哪个 session 与。有没有办法通过使用 tf.Print() 或其他调试方法来解决这个问题？谢谢!

Answer 1

tf.Print用作恒等函数，它返回您作为第一个参数传递的相同张量，具有打印指定为第二个参数的张量列表的副作用。

所以你应该使用如下:

def loss(a, b):
    ...
    a = tf.Print(a, [a], message = 'debug: ')
    return tf.add(a, b)

a 将在每次评估张量 tf.add(a, b) 时打印。