Enable other axes support: the take test

[mozga-intel]

Description

There is an axis param that is set up in a loop: [loop] Loop control variable 'axis' not used within the loop body but it's only used out of the loop scope: loop

    for axis in range(-data_ndim, data_ndim):
        data_shape = ()
        for _ in range(data_ndim):
            data_shape += (np.random.randint(low=1, high=5), )
        idx_shape = ()
        for _ in range(idx_ndim):
            idx_shape += (np.random.randint(low=1, high=5), )
     ....
    
    result = mx.sym.take(a=data, indices=idx, axis=axis, mode=mode)

The axis is always equal to data_ndim- 1. Following the documentation doc, the axis of an input array is to be taken. For the input tensor of rank r, it could be in the range of [-r, r-1]. Where r = data_ndim;

Checklist

Essentials

• PR's title starts with a category (e.g. [BUGFIX], [MODEL], [TUTORIAL], [FEATURE], [DOC], etc)
• Changes are complete (i.e. I finished coding on this PR)
• All changes have test coverage
• Code is well-documented

Changes

• Feature1, tests, (and when applicable, API doc)
• Feature2, tests, (and when applicable, API doc)

Comments

• If this change is a backward incompatible change, why must this change be made.
• Interesting edge cases to note here

[mxnet-bot]

Hey @mozga-intel , Thanks for submitting the PR
All tests are already queued to run once. If tests fail, you can trigger one or more tests again with the following commands:

• To trigger all jobs: @mxnet-bot run ci [all]
• To trigger specific jobs: @mxnet-bot run ci [job1, job2]

---

CI supported jobs: [windows-cpu, unix-cpu, edge, unix-gpu, miscellaneous, clang, windows-gpu, website, centos-gpu, sanity, centos-cpu]

---

Note:
Only following 3 categories can trigger CI :PR Author, MXNet Committer, Jenkins Admin.
All CI tests must pass before the PR can be merged.