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

mac2022-06-30  61

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

转载于:https://www.cnblogs.com/shixisheng/p/9460614.html

最新回复(0)