如何在tensorflow中打印完整(不截断)张量？

Question

每当我尝试打印时,我总会得到截断的结果

import tensorflow as tf
import numpy as np

np.set_printoptions(threshold=np.nan)

tensor = tf.constant(np.ones(999))

tensor = tf.Print(tensor, [tensor])

sess = tf.Session()

sess.run(tensor)

正如你所看到的,我遵循了一个指南,我发现打印完全值张量到控制台或写入张量流文件

但输出很简单

...\core\kernels\logging_ops.cc:79] [1 1 1 ...]

我想看到完整的张量,谢谢.

Answer 1

您可以在 TensorFlow 2.x 中执行以下操作：

import tensorflow as tf

tensor = tf.constant(np.ones(999))
tf.print(tensor, summarize=-1)

来自 TensorFlow 文档 -> 总结：每个维度中的第一个和最后一个总结元素是按张量递归打印的。如果设置为 -1，它将打印每个张量的所有元素。

https://www.tensorflow.org/api_docs/python/tf/print

Answer 2

通过检查Tensorflow API可以轻松解决这个问题tf.Print.通过summarize=n这里n是你想要显示的元素的数量.