numpy - 如何在 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)

如您所见，我遵循了在 Print full value of tensor into console or write to file in tensorflow 上找到的指南

但是输出很简单

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

我想查看完整的张量，谢谢。

Answer 1

通过检查 the Tensorflow API 可以轻松解决此问题对于tf.Print。传递 summarize=n，其中 n 是要显示的元素数量。