pytorch

ea6113ea - Update loss.py (#95367)

Update loss.py (#95367) Fix the dimension bug in the document Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/95367 Approved by: https://github.com/albanD, https://github.com/kit1980