首页 > 其他 > 详细

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

时间:2018-08-11 19:13:00      阅读:1704      评论:0      收藏:0      [点我收藏+]

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

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

原文:https://www.cnblogs.com/shixisheng/p/9460614.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!