Features/1383 Add vmap functionality for vectorization across split dimension

[mrfh92]

Due Diligence

• General:
  • [x] title of the PR is suitable to appear in the Release Notes
• Implementation:
  • [x] unit tests: all split configurations tested
  • [x] unit tests: multiple dtypes tested
  • [x] documentation updated where needed

Description

Issue/s resolved: #1383 The implemented vmap-functionality is similar to the one of PyTorch. It allows vectorized application of PyTorch-functions to one or multiple DNDarrays with vectorization taking place along the respective split axes.

An artificial example, how vmap does work, is found in the tests.

Type of change

new feature

Does this change modify the behaviour of other functions? If so, which?

no