Fix ONNX LogSoftmax export.#9576
Fix ONNX LogSoftmax export.#9576kit1980 wants to merge 1 commit intopytorch:masterfrom kit1980:master
Conversation
|
Now one test is failing on exporting LogSoftmax model, but without this change it exports to a wrong ONNX model. |
|
#9660 will fix the wrong test cases. |
|
#9660 is in now, how can we proceed with this PR? |
facebook-github-bot
left a comment
There was a problem hiding this comment.
houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
|
@pytorchbot retest this please |
|
@kit1980 can you rebase the code to the master? |
|
@houseroad rebased on master. |
facebook-github-bot
left a comment
There was a problem hiding this comment.
houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
facebook-github-bot
left a comment
There was a problem hiding this comment.
houseroad has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
Summary: This fixes an issue with incorrect `axis=-1` in the exported ONNX. Pull Request resolved: pytorch#9576 Reviewed By: yinghai Differential Revision: D9125463 Pulled By: houseroad fbshipit-source-id: 6f4cb1067d1aa6bb0a9f56690fc21816c98eebfa
This fixes an issue with incorrect
axis=-1in the exported ONNX.