pytorch
944ea4c3 - ONNX Export Support for CrossEntropyLoss (#33767)

Commit
4 years ago
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
Author
Parents
Loading