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

时间：2018-08-11 19:33:37 收藏：0 阅读：6385

使用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_ 相同的张量。