TensorFlow打印tensor值的实现方法
最近一直在用TF做CNN的图像分类，当softmax层得到预测结果后，我希望能够看到预测结果，以便和标签之间进行比较。特此补上，以便自己记忆。
我现在通过softmax层得到变量train_logits，如果我直接执行print(train_logits)时，得到的结果如下（因为我是134类分类，所以结果是（1,134）维）：
这貌似什么都看不出来。
其实tensorflow提供输出中间值方法方便debug。
这个函数就是[tf.Print]。
tf.Print( input_, data, message=None, first_n=None, summarize=None, name=None )
参数:
input_: 通过此op的一个tensor.
data: 当此op被计算之后打印输出的tensor list。
message: 错误消息的前缀，是一个string。
first_n: 只记录first_n次. 总是记录负数;这是个缺省.
summarize: 对每个tensor只打印的条目数量。如果是None，对于每个输入tensor只打印3个元素。
name: op的名字.
返回值:
和input_相同的tensor.
将print(train_logits)修改为sess.run(tf.Print(train_logits,[train_logits]))后，得到的结果如下图：
发现值已经能够打印啦！但是只能输出3个条目，这是因为参数summarize默认只打印3个参数！
将sess.run(tf.Print(train_logits,[train_logits]))修改为sess.run(tf.Print(train_logits,[train_logits],summarize=134))后（我是134类分类），得到的结果如下图：
这样就能成功地打印tensor的中间结果了！以后调参、查看模型效果就直观多了！
以上这篇TensorFlow打印tensor值的实现方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了使用Python刷淘宝喵币（低阶入门版）的相关资料,非常不错，具有一定的参考借鉴价值，需要的朋友可以参考下2019-10-10
实习项目要做安全帽目标检测，拿到了公司给的一些视频数据，使用Opencv读取视频并每隔1s存储一副图像，本文就详细的介绍一下使用，感兴趣的可以了解一下2021-06-06
这篇文章主要为大家详细介绍了python tkinter库实现气泡屏保和锁屏，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-07-07
在Python中,可以使用多种方法来多次遍历文件,下面这篇文章主要给大家介绍了关于python遍历文件夹的各种方法,文中通过代码介绍的非常详细,需要的朋友可以参考下2024-03-03
这篇文章主要介绍了详解Python函数可变参数定义及其参数传递方式的相关资料,这里提供实例代码帮助大家学习理解这部分内容，需要的朋友可以参考下2017-08-08
这篇文章主要介绍了python常用函数random()函数,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-02-02
这篇文章主要介绍了Python使用socket_TCP实现小文件下载功能,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2020-10-10
这篇文章主要介绍了Python深度学习关于pytorch神经网络图像卷积的运算示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助2021-10-10
Python3.6 + TensorFlow 安装配置图文教程（Windows 64 bit）
这篇文章主要介绍了Python3.6 + TensorFlow 安装配置的教程（Windows 64 bit）,本文通过图文并茂的形式给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下2020-02-02
这篇文章主要介绍了python去除空格,tab制表符和\n换行符的小技巧，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-01-01
最新评论