chunked/nltk_chunking/tasks_document_question_answering/chunk_105.txt

Take an argmax on the last dimension of both the start_logits and end_logits to get the predicted start_idx and end_idx.