multidim-indexing (pypi): Multidimensional batch indexing of pytorch tensors and numpy arrays
Keywords: numpy|pytorch|indexing|arrays|batch|batch-processing|python|tensor|voxel-grid|voxels