ONNX Export Support for CrossEntropyLoss (#33767)
Summary:
Add ONNX export support for torch.nn.CrossEntropyLoss.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/33767
Reviewed By: hl475
Differential Revision: D20121169
Pulled By: houseroad
fbshipit-source-id: 7b56617e8c60617b922949fc8b4ecc626eedf7ed