I have a CNN model with two inputs, like this:
input_a = tf.keras.layers.Input(shape=[128, 128, 3]) input_b = tf.keras.laye