Implement take_along_axis function per Python Array API

[oleksandr-pavlyk]

The function is planned for Python Array API 2024.12 specification.

• Have you provided a meaningful PR description?
• Have you added a test, reproducer or referred to an issue with a reproducer?
• Have you tested your changes locally for CPU and GPU devices?
• Have you made sure that new changes do not introduce compiler warnings?
• Have you checked performance impact of proposed changes?
• If this PR is a work in progress, are you opening the PR as a draft?

[github-actions]

Deleted rendered PR docs from intelpython.github.com/dpctl, latest should be updated shortly. 🤞

[coveralls]

coverage: 87.927% (+0.2%) from 87.739%
when pulling d07de32 on take-along-axis
into d79dae1 on master.

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_205 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_208 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_209 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_235 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_237 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_239 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_240 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_245 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_245 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[github-actions]

Array API standard conformance tests for dpctl=0.18.0dev0=py310ha798474_246 ran successfully.
Passed: 894
Failed: 1
Skipped: 119

[antonwolfy]

I've opened a PR to reuse the new dpctl function in dpnp-1969. The existing and new dpnp tests are passed with that dpctl PR locally.

[ndgrigorian]

@oleksandr-pavlyk
I am going to try to take a look at this today.

One thing that I did realize, though: would it make sense to add put_along_axis as an inverse function? Array API obviously won't yet due to mutation issues, but we already have put, it seems like a good idea, maybe for a separate PR.

[oleksandr-pavlyk]

One thing that I did realize, though: would it make sense to add put_along_axis as an inverse function?

Yes, makes total sense. Let's do it in a separate PR though.

[ndgrigorian]

I've tested the branch out and it works, this LGTM.

Thank you @oleksandr-pavlyk !