pytorch
4c9d6607 - fix gather issue when index is shape of n by 1 (#99709)

Commit
2 years ago
fix gather issue when index is shape of n by 1 (#99709) Fix https://github.com/pytorch/pytorch/issues/99595 When the index is shape of {N, 1}, it will also have strides of {1, 0}, which is the same as an expanded tensor (e.g. shape of {5, 5} and strides {1, 0}), leading to wrong output. Pull Request resolved: https://github.com/pytorch/pytorch/pull/99709 Approved by: https://github.com/XiaobingSuper, https://github.com/ezyang
Author
Committer
Parents
Loading