projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e2730dd) | patch
Support dim=None for argmax and argmin (#18264)
authorXiang Gao <qasdfgtyuiop@gmail.com>
Tue, 26 Mar 2019 03:36:44 +0000 (20:36 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 26 Mar 2019 03:43:34 +0000 (20:43 -0700)
commitbf2a30cb22afe88b696b41edad6c2c147f9425f2
tree7936a0e3ddb3f24f656789d2c8ed685527e082d1tree | snapshot
parente2730ddb21164f1cb36330289ab2cd0629c06dc8commit | diff
Support dim=None for argmax and argmin (#18264)

Summary:
Fixes: https://github.com/pytorch/pytorch/issues/18263
cc: houseroad
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18264

Reviewed By: ezyang

Differential Revision: D14559234

Pulled By: houseroad

fbshipit-source-id: c5b8623752d6c6af41c6d715fd9585a65294868d
test/onnx/test_pytorch_onnx_caffe2.py diff | blob | history
torch/onnx/symbolic.py diff | blob | history
Domain: Machine Learning / ML Framework;