def METHOD_NAME(out):
    ans = []
    for batch_id in range(out.voxel_batch_splits.shape[0] - 1):
        start = out.voxel_batch_splits[batch_id]
        end = out.voxel_batch_splits[batch_id + 1]
        voxels = {}
        for i, c in enumerate(out.voxel_coords[start:end]):
            point_indices = out.voxel_point_indices[out.voxel_point_row_splits[
                start + i]:out.voxel_point_row_splits[start + i + 1]]
            voxels[tuple(c)] = set(point_indices)
        ans.append(voxels)
    return ans