-
Notifications
You must be signed in to change notification settings - Fork 6.8k
moveaxis operator now accepts negative indices and sequence of ints as well. #14321
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It would be good to have it as a symbol operator as well to be able to use it in hybridized graphs rather than just ndarray operations.
@mxnet-label-bot add [pr-awaiting-merge] |
@ifeherva could you check the CI failures and re-trigger the build? |
@vandanavk the error messages do not really make sense to me, they look unrelated to the content of this PR. Can someone retrigger CI please? |
Tests are very flaky, none of them related to this PR. |
Sorry that I wrote a typo, which should be retrigger. |
I do not know why the CI always fail in this PR. |
Strange that the same GPU tests fail though I am puzzled how these changes contribute to CUDA errors. |
Could we disable the new testcase to track the error? |
|
Current MXNet Does not support zero size tensor, so I modified the testcase. |
Great! Glad to merge it. |
Thanks for figuring out the error! |
…s well. (apache#14321) * nd.moveaxis now accepts negative indices and sequence of ints as well. * Retrigger CI * retrigger CI * Retriever * Retrigger CI * Retrigger CI * Retrigger CI, hope for success * Update test_ndarray.py * Let’s try to retrigger CI again. * Retrigger CI * Disable the new testcase to track bugs * enable some new testcases * Update test_ndarray.py * Update test_ndarray.py
…s well. (#14321) * nd.moveaxis now accepts negative indices and sequence of ints as well. * Retrigger CI * retrigger CI * Retriever * Retrigger CI * Retrigger CI * Retrigger CI, hope for success * Update test_ndarray.py * Let’s try to retrigger CI again. * Retrigger CI * Disable the new testcase to track bugs * enable some new testcases * Update test_ndarray.py * Update test_ndarray.py
…s well. (apache#14321) * nd.moveaxis now accepts negative indices and sequence of ints as well. * Retrigger CI * retrigger CI * Retriever * Retrigger CI * Retrigger CI * Retrigger CI, hope for success * Update test_ndarray.py * Let’s try to retrigger CI again. * Retrigger CI * Disable the new testcase to track bugs * enable some new testcases * Update test_ndarray.py * Update test_ndarray.py
Description
The moveaxis operator is now numpy compatible by accepting negative indices and tuples as well. Added a series of unit tests based on the original numpy repo.
Adds request from #13563
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.
Changes are complete (i.e. I finished coding on this PR)
All changes have test coverage:
Unit tests are added for small changes to verify correctness (e.g. adding a new operator)
Code is well-documented:
For user-facing API changes, API doc string has been updated.
To the my best knowledge, examples are either not affected by this change, or have been fixed to be compatible with this change