Доступ к промежуточному тензору тензорного графа Кераса TF2 - PullRequest
0 голосов
/ 07 августа 2020

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

Спасибо!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...