首页 > 解决方案 > 访问 TF2 keras 张量图的中间张量

问题描述

是否可以访问TensorTensorFlow 2 图中的中间对象?通过这个,采取以下简短的例子:

x1 = tf.keras.Input(shape=(1,))
x2 = tf.keras.layers.Dense(1,name="l1")(x1)
x3 = tf.keras.layers.Dense(2,name="l2")(x2)
x4 = tf.keras.layers.Dense(1,name="l3)(x3)
model = tf.keras.Model(inputs=x1,outputs=x4)

other_x = tf.keras.Input(shape=(1,))
model_out = model(other_x)

model_out这是一个Tensor对象(它的形状:[None,1])。我想访问例如在评估第二个密集层(“l2”层的输出)之后的中间张量。我相信这个张量被埋在 中的某个地方model_out.graph,但我不确定提取它的最佳方法是什么?我不想建立一个单独的模型,因为在我的实际情况下有一些随机(随机)层,所以我确实想得到图中的中间张量。

谢谢!

标签: tensorflow2.0tf.keras

解决方案


推荐阅读