解决TensorFlow打印Tensor时出现省略号的问题
在TensorFlow调试过程中,打印大型Tensor时常因默认限制显示为省略号。本文介绍通过导入NumPy模块并调用set_printoptions(threshold=np.inf)函数,将打印阈值设置为无限大,从而完整输出Tensor所有数据,解决显示不全的问题。
TensorFlow打印Tensor省略号问题及解决方案
首先展示相关代码示例:
import tensorflow as tf
x = tf.ones(shape=[100, 200], dtype=tf.int32, name='x')
y = tf.zeros(shape=[2, 3], dtype=tf.float32, name='y')
with tf.Session() as sess:
print(sess.run([x, y]))
运行上述代码后,得到的输出结果如下所示:

如果在调试阶段希望查看被省略号隐藏的具体数值内容,
只需对代码进行如下调整:
import tensorflow as tf
import numpy as np #借助numpy模块的set_printoptions()函数,将打印上限设置为无限即可
np.set_printoptions(threshold=np.inf)
x = tf.ones(shape=[10, 20], dtype=tf.int32, name='x')
y = tf.zeros(shape=[2, 3], dtype=tf.float32, name='y')
with tf.Session() as sess:
print(sess.run(x, y))
再次运行后的输出结果如上图所示,此时可以清晰地看到原本被省略的部分数据也已完整打印出来。

以上即为解决TensorFlow中打印Tensor出现省略号问题的完整方法,希望能对各位开发者有所帮助,也欢迎大家继续支持本站获取更多技术分享。
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









