tf.keras.Model中的Tensorflow可选输入

Question

我正在尝试建立一个模型，该模型在培训阶段将可选输入传递给某些层。可以使用tf.keras.Model来做到这一点吗？

基本上，某些层具有如下所示的调用例程：

def call(self, inputs, optional_inputs=None, training=None)
    if training:
       assert optional_inputs is not None
       return funA(inputs, optional_inputs)
    return funB(inputs)

是否可以这样做而不必从头开始实现所有model.fit等？