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