Maybe can try:
x = torch.randint(0, 10, (10,3))

splits = torch.tensor([0, 2, 3,6,7,10])

x_split = [x[splits[i]:splits[i+1]] for i in range(len(splits)-1)]

Or something to that effect, depending on the specifics.