Можно ли получить доступ к промежуточному объекту Tensor
на графике TensorFlow 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
, но я не уверен, как лучше всего его извлечь? Я не хочу строить отдельную модель, так как в моем реальном случае есть некоторые сточасти c (случайные) слои, поэтому я действительно хочу получить промежуточный тензор на графике.
Спасибо!