使用tf.print()打印tensor内容

Posted 2021-01-02 shixisheng

使用tf.Print()打印tensor内容，这是tensorflow中调试bug的一个手段，例子如下所示：

import tensorflow as tf

a = tf.Variable(tf.random_normal([3, 3, 1, 64], stddev=0.1))
a = tf.Print(a, [a], "a: ",summarize=9)
init = tf.global_variables_initializer()
sess = tf.Session()
sess.run(init)
sess.run(a)

输出结果

tf.Print()函数参数为：

Print(
    input_,
    data,
    message=None,
    first_n=None,
    summarize=None,
    name=None
)

参数：

• input_：通过这个操作的张量。
• data：计算 op 时要打印的张量列表。
• message：一个字符串，错误消息的前缀。
• first_n：只记录 first_n 次数。负数日志，这是默认的。 
• summarize：只打印每个张量的许多条目。如果没有，则每个输入张量最多打印3个元素。
• name：操作的名称（可选）。

返回：

该操作将返回与 input_ 相同的张量。