pytorch
2c57bbf5 - add support for indexing to meshgrid (#62722)

Commit
3 years ago
add support for indexing to meshgrid (#62722) Summary: This is step 3/7 of https://github.com/pytorch/pytorch/issues/50276. It only adds support for the argument but doesn't implement new indexing modes yet. Pull Request resolved: https://github.com/pytorch/pytorch/pull/62722 Test Plan: Verified this is not FC breaking by adding logging to both meshgrid overloads and then called meshgrid twice: `meshgrid(*tensors)` and `meshgrid(*tensors, indexing='ij')` This confirmed that the former signature triggered the original native function and the latter signature triggered the new native function. Reviewed By: H-Huang Differential Revision: D30394313 Pulled By: dagitses fbshipit-source-id: e265cb114d8caae414ee2305dc463b34fdb57fa6
Author
Michael Dagitses
Parents
Loading