I wanted to extract and apply independently a Conv2D layer on the columns of my input tensors, but after adding the code:
accelerometer_input = Input(shape=(1400, 3))
for i in range(3):
out = Lambda(lambda x: x[:,:, i:i+1])(accelerometer_input) # Extracting the ith channel
out = K.expand_dims(out, axis=1)
out = Conv2D(64, (30, 1), data_format="channels_first")(out)
branch_outputs.append(out)
out_put = K.concatenate(branch_outputs)
it gives me the error in the title. I think it is due to the Lambda layer or the extraction which is not differentiable.
But how can I do without it?