InvalidArgumentError: assertion failed: [Condition x == y did not hold element-wise:]

Question

I am trying to do a classification with LSTM (to do event prediction) I have 320 classes. The dataset is made up of time series data and was created and trained using the follow