assertion failed: [Condition x == y did not hold element-wise:] Function call stack: train_function

Question

hi i have the following code

model = tf.keras.Sequential()
model.add(tf.keras.layers.LSTM(64, return_sequences=True, recurrent_regularizer=l2(0.0015), 
input_shap