时间戳

首页 > 解决方案 > [Tensorflow 2.0][张量值打印]

python - [Tensorflow 2.0][张量值打印]

问题描述

我们最近将我们的 Tensorflow1.12.0 源代码移植到了 Tensorflow2.0。在调试期间，我们想打印张量的值。例如，考虑以下代码片段：

 def __build_model():
        input = tf.keras.Input(shape=(None, self.__input_size), name='inputs')
        dense_layer_output = tf.keras.layers.Dense(self.__output_size)(next_input)
        self.__model = tf.keras.Model(inputs=input, outputs=dense_layer_output)

为了打印张量input的值，我们只是根据 [1]使用tf.print(input)函数打印了变量（以前只是print(input)）。因此，上面的代码片段可以重写如下：

def __build_model():
        input = tf.keras.Input(shape=(None, self.__input_size), name='inputs')
        dense_layer_output = tf.keras.layers.Dense(self.__output_size)(next_input)
        tf.print(input)
        self.__model = tf.keras.Model(inputs=input, outputs=dense_layer_output)

但是，我们观察到的输出是张量对象，而不是输入张量的确切值。谁能帮我用语法来打印确切的张量值，而不是张量对象。

[1] https://www.tensorflow.org/api_docs/python/tf/print?version=stable

谢谢。

标签： pythontensorflowtensorflow2.0

解决方案

推荐阅读