def multiply(x): return tf.math.multiply(x=x[0, :], y= x[1, 0]) inputs = Input(shape=(2, )) r= Dense(units=1, activation=multiply)(inputs) fr = Dense(units=1, activation='relu', )(r) ...
When running this code I get the following error:
ValueError: Input 0 of layer dense_38 is incompatible with the layer: : expected min_ndim=2, found ndim=1. Full shape received: (1,)
I do not understand why it is expecting a two dimensional input