首页 > 解决方案 > 由于 Keras 变量名称未出现在摘要中,如何通过名称访问回调或其他地方的 Keras 变量?

问题描述

这是在 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?

标签: kerastensorflow2.0

解决方案


推荐阅读