这是在tensorflow-2.0 keras中。
x_input keras.layers.Input(shape=(1,))
V = K.variable(0.5, name='V', dtype=tf.float32)
V = tf.reduce_mean(x_input, axis=-1) * 0 + V # this is a stupid way to get things to work
V = tf.expand_dims(V, axis=-1)
model = keras.models.Model(inputs=x_input, outputs=V)
然后,我将此模型(未编译)用作另一个模型构建过程的输入。
第二个模型中的第一个模型的摘要中都没有显示变量“ V”。如何在回调甚至急切模式中保持值V?