categorical_cross_entropy in tensorflow v1: ValueError: Rank mismatch: Rank of labels (received 2) should equal rank of logits minus 1 (received 2)

Question

I am trying to convert a question answering model which generates a single word as an answer to a multiple-word answer. The loss calculation of single word answer model is a