在Tensorflow上提取图层的输出

Question

如何在Tensorflow中获取单个图层的输出张量。 说完TF模型输入图像通过之后，我想按名称获取某些图层的输出。

我怎么能得到它？

Answer 1

假设您使用name="output"为图层命名，则可以使用以下命令获得特定的输出。

output = graph.get_tensor_by_name('output:0')

其中graph是使用graph = tf.get_default_graph()获得的默认图形。但是，请注意output是张量。我想您想对输出进行一些操作/可视化。在这种情况下，您需要在Session中运行此命令。像这样：

with tf.Session() as sess:
    output = graph.get_tensor_by_name('output:0')
    output_values = sess.run(output, feed_dict={x: input})