tensorflow实现在函数中用tf.Print输出中间值
tensorflow由于其基于静态图的模式，导致写代码的时候很难调试，除了用官方的调试工具外，最直接的方法就是把中间结果输出出来查看，然而，直接用print函数只能输出tensor变量的形状，而不是数值，想要输出tensor的具体数值需要用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：操作的名称（可选）
然而网上大部分资源都是介绍如何在主函数中先建立一个op，再开启一个Session执行sess.run(op)的方法，但是如果想要输出函数中的中间值而该值又未传回主函数呢？这种情况下无法在函数中开启一个新的Session，但是仍然可以用tf.Print建立op来实现。
import tensorflow as tf import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" def test(): a=tf.constant(0) for i in range(10): a_print = tf.Print(a,['a_value: ',a]) a=a_print+1 return a if __name__=='__main__': with tf.Session() as sess: sess.run(test())
运行结果：
a_print可以理解为在图中新增了一个节点，在后续代码中当有别的变量使用了a_print时（如上例a=a_print+1），就会有数据从a_print节点上流过，就会输出值，而究竟会输出几次值呢？这其实并不是看下文中a_print被使用了几次，而是看数据流要从该节点上流经几次，可以理解为a_print这个op被“定义”了几次。
def test(): a=tf.constant(0) a_print = tf.Print(a,['a_value: ',a]) for i in range(10): a=a_print+1 return a if __name__=='__main__': with tf.Session() as sess: sess.run(test())
如果把test()函数改成这样，则运行结果为：
输出仅被执行了一次，因为a_print这个op只被定义了一次，虽然后面在循环里不断被a使用，但是数据只从它身上经过了一次，所以只会print一次，并且a_print的值永远为0，最终返回的a的值也为1。
再把代码改成下例：
def test(): a=tf.constant(0) a_print = tf.Print(a,['a_value: ',a]) for i in range(10): a_print=a_print+1 return a if __name__=='__main__': with tf.Session() as sess: sess.run(test())
运行结果是什么也不会输出，因为a_print这个op没有和别的变量发生关系，它没有被别的变量使用，在图里为孤立的一个节点，没有数据流过，就不会被执行。
而如果改成这样
def test(): a=tf.constant(0) a_print = tf.Print(a,['a_value: ',a]) for i in range(10): a_print=a_print+1 return a_print if __name__=='__main__': with tf.Session() as sess: sess.run(test())
运行结果
返回的a_print值为10也是正确的，因为a_print在下文被返回，所以有数据流流经，会被执行，而因为a_print的定义只执行一次，所以只会输出一次。
以上这篇tensorflow实现在函数中用tf.Print输出中间值就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了用十张图详解TensorFlow数据读取机制（附代码），小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-02-02
正则表达式是一个特殊的字符序列，它能帮助你方便的检查一个字符串是否与某种模式匹配。这篇文章主要介绍了python使用正则表达式替换匹配成功的组并输出替换的次数,需要的朋友可以参考下2017-11-11
python+selenium 简易地疫情信息自动打卡签到功能的实现代码
这篇文章主要介绍了python+selenium 简易地疫情信息自动打卡签到功能的实现代码,代码简单易懂，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-08-08
python包pdfkit(wkhtmltopdf) 将HTML转换为PDF的操作方法
pdfkit，把HTML+CSS格式的文件转换成PDF格式文档的一种工具。它就是html转成pdf工具包wkhtmltopdf的Python封装。所以，必须手动安装wkhtmltopdf,这篇文章主要介绍了python包pdfkit(wkhtmltopdf)将HTML转换为PDF,需要的朋友可以参考下2022-04-04
LyScript是一款x64dbg主动化操控插件，经过Python操控X64dbg，完成了远程动态调试，解决了逆向工作者剖析漏洞，寻觅指令片段，原生脚本不行强壮的问题。本文将利用LyScript插件实现批量打开关闭进程，感兴趣的可以了解一下2022-07-07
Python判断对象是否为文件对象(file object)的三种方法示例
这篇文章主要介绍了Python判断对象是否为文件对象(file object)的三种方法示例，https://www.pythontab.com/html/2018/pythonhexinbiancheng_1015/1362.html2019-04-04
在日常工作中,文件对象是我们常接触到的可迭代类型之一,一般用 for 循环遍历一个文件对象,可以逐行读取它的内容,但这种方式在碰到大文件时,可能会出现一些奇怪的效率问题,所以本文给大家介绍了Python3读取和处理超大文件的操作,需要的朋友可以参考下2024-04-04
柯里化（Currying）是函数式编程中的一个重要概念,它可以将一个多参数函数转化为一系列单参数函数的组合,本文将详细解释什么是柯里化,如何在Python中实现柯里化,感兴趣的可以了解下2024-01-01
这篇文章主要介绍了python里使用正则表达式的后向搜索肯定模式,本文通过代码介绍的非常详细，包括语法介绍，非常不错，具有参考借鉴价值，需要的朋友可以参考下2017-11-11
正则表达是用来匹配字符串，这篇文章给大家介绍了python正则表达式及正则表达式的例子，文章给大家提到了正则表达式语法规则，感兴趣的朋友一起看看吧2018-01-01
最新评论