I remember this transpose is necessary since our op follows MXNet convention. gather, gather_nd in MXNet have different expectation on indices argument compared to other framework.
I remember this transpose is necessary since our op follows MXNet convention. gather, gather_nd in MXNet have different expectation on indices argument compared to other framework.