前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >tf.Print()

tf.Print()

作者头像
狼啸风云
修改2022-09-03 22:01:06
5K0
修改2022-09-03 22:01:06
举报

调试程序的时候,经常会需要检查中间的参数,这些参数一般是定义在model或是别的函数中的局部参数,由于tensorflow要求先构建计算图再运算的机制,也不能定义后直接print出来。tensorflow有一个函数tf.Print()。

代码语言:javascript
复制
tf.Print(input, data, message=None, first_n=None, summarize=None, name=None)

最低要求两个输入,input和data,input是需要打印的变量的名字,data要求是一个list,里面包含要打印的内容。

参数:

  • message是需要输出的错误信息
  • first_n指只记录前n次
  • summarize是对每个tensor只打印的条目数量,如果是None,对于每个输入tensor只打印3个元素
  • name是op的名字

需要注意的是tf.Print()只是构建一个op,需要run之后才会打印。

例子:

代码语言:javascript
复制
x=tf.constant([2,3,4,5])
x=tf.Print(x,[x,x.shape,'test', x],message='Debug message:',summarize=100)
     
with tf.Session() as sess:
    sess.run(x)
     
#Debug message:[2 3 4 5][4][test][2 3 4 5]
     
x=tf.Print(x,[x,x.shape,'test', x],message='Debug message:',summarize=2)
     
with tf.Session() as sess:
    sess.run(x)
     
#Debug message:[2 3...][4][test][2 3...]

输出是在命令窗口中,和print有区别

用print也可以输出:

代码语言:javascript
复制
x=tf.constant([2,3,4,5])
     
with tf.Session() as sess:
    print(sess.run(x))
     
#[2,3,4,5]

如果希望打印的参数是一个中间参数,换言之是一个局部参数不会被传回主函数。那么首先需要定义一个全局参数去保存它,一般在函数就是定义在最前面的self.para = para,再在主函数中print出来。

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019年09月02日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档