[DOC] Typo and fix the input of labels to `cross_entropy`
# What does this PR do?
The current version caused some errors. The proposed changes fixed it for me. Hope this is helpful!

- [ x] This PR fixes a typo or improves the docs (you can dimiss the other checks if that's the case).


@sgugger 