Я подклассифицирую tf.keras.Model и пытаюсь выяснить, как использовать другой слой для вывода во время вывода. Как ты это делаешь? Если бы это было с функциональным API, вы могли бы сделать это следующим образом.
new_output = model.get_layer(LAYER_NAME).output
new_input = tf.keras.Input([...])
new_model = tf.keras.models.Model(inputs=new_input, outputs=new_ouput)
y = new_model.predict(x)