一尘不染

Python-如何在TensorFlow中打印Tensor对象的值?

python

我一直在使用TensorFlow中矩阵乘法的入门示例。

matrix1 = tf.constant([[3., 3.]])
matrix2 = tf.constant([[2.],[2.]])
product = tf.matmul(matrix1, matrix2)

当我打印产品时,它显示为一个Tensor对象:

<tensorflow.python.framework.ops.Tensor object at 0x10470fcd0>

但是我怎么知道价值product呢?

以下内容无济于事:

print product
Tensor("MatMul:0", shape=TensorShape([Dimension(1), Dimension(1)]), dtype=float32)

我知道图形可以继续运行Sessions,但是没有任何方法可以在Tensor不运行图形的情况下检查对象的输出session吗?


阅读 1986

收藏
2020-02-17

共1个答案

一尘不染

评估对象实际值的最简单方法[A]Tensor是将其传递给Session.run()方法,或者Tensor.eval()在有默认会话时(即在一个with tf.Session():块中,或参阅下文)调用该方法。通常[B],如果不在会话中运行某些代码,就无法打印张量的值。

如果你正在试验编程模型,并且想要一种简单的方法来评估张量,则tf.InteractiveSession可以让你在程序开始时打开一个会话,然后将该会话用于所有Tensor.eval()(和Operation.run())调用。当在一个Session无处不在的对象周围传递乏味的代码时,在诸如外壳或IPython笔记本之类的交互式设置中,这可能会更容易。例如,以下内容在Jupyter笔记本中起作用:

with tf.Session() as sess:  print(product.eval()) 

对于这么小的表达式来说,这似乎很愚蠢,但是Tensorflow中的关键思想之一是推迟执行:构建一个大型而复杂的表达式非常便宜,并且当你要对其进行评估时,可以使用后端(连接到该表达式)使用Session)能够更有效地安排其执行时间(例如,并行执行独立部分并使用GPU)。

[A]:要打印张量的值而不将其返回到Python程序,可以使用tf.Print()运算符,如Andrzej在另一个答案中建议的那样。请注意,你仍然需要运行部分图形来查看此op的输出,该输出已打印到标准输出中。如果你正在运行分布式TensorFlowtf.Print()则将其输出打印到该op运行所在任务的标准输出中。这意味着,例如,如果你使用https://colab.research.google.com或任何其他Jupyter Notebook,则你在笔记本中将看不到输出tf.Print();在这种情况下,请参考此答案以获取如何使其仍然打印。

[B]:你也许可以使用实验性tf.contrib.util.constant_value()函数来获取恒定张量的值,但它并不打算用于一般用途,也没有为许多运算符定义。

2020-02-17