Expand basis functions to operate on arbitrary dimensions

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 90.00%. Comparing base (39aed1e) to head (48e16d7).

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #38 +/- ## ========================================== + Coverage 89.59% 90.00% +0.40% ========================================== Files 21 21 Lines 298 310 +12 ========================================== + Hits 267 279 +12 Misses 31 31 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.