I’m not sure to understand the use case completely.
Could you post some examples using smaller tensor shapes, e.g. [1, 3, 10], and demonstrate how you would like to split this tensor using the indices?