Как вывести скалярную переменную как вывод модели Keras?

Question

Например, я хочу что-то вроде

Модель (входы = что-то, выходы = скаляр)

Это происходит, когда вы хотите отладить модель / процедуру обучения в случае снет состояния "X" первым (генеративным).Таким образом, у вас все еще есть batch_size от Y. И это то, что вы хотите.

Я пытаюсь что-то вроде этого:

V = K.variable(0, dtype=tf.float32)
V = tf.reduce_mean(x_input_not_used_by_this_branch, axis=1) * 0 + V # this is a stupid way to get things to work
model keras.models.Model(inputs=something, outputs=[V, some_other_stuff])

Answer 1

Простой способ - использовать функциональный API от Keras: Документация по Keras API

inputs = Input(shape=(784,))

# a layer instance is callable on a tensor, and returns a tensor
x = Dense(64, activation='relu')(inputs)
x = Dense(64, activation='relu')(x)
predictions = Dense(10, activation='softmax')(x)

v = 0.25 * x

# This creates a model that includes
# the Input layer and three Dense layers
model = Model(inputs=inputs, outputs=[predictions, v])