Multidimensional array batch indexing for pytorch and numpy

Follow the full discussion on Reddit.
Batch indexing into multidimensional tensors/arrays is kind of tricky, I made this project explaining the builtin syntax and also made wrappers for simplifying the interface, with additional features for underlying coordinate grid data (like signed distance functions) that need to be indexed by coordinate value rather than integer indices directly https://github.com/LemonPi/multidim_indexing