如何在张量流中设置keras子类模型的输入？

Question

我已经使用tensorflow创建了一个keras子类模型。片段如下所示。

class SubModel(Model):
    def call(self, inputs):
        print(inputs)

model = SubModel()
model.fit(data, labels, ...)

当模型为fit时，它将获得输入和input_shape本身。我想做的就是自己将输入传递给模型。就像功能API一样。

inputs = tf.keras.input(shape=(100,))
model = tf.keras.Model(inputs=inputs, outputs=outputs)

Answer 1

类似的东西？

model_ = SubModel()
inputs = tf.keras.input(shape=(100,))
outputs = model_(inputs)
model = tf.keras.Model(inputs=inputs, outputs=outputs)