Hi all,
Looking for the exact functionality provided by torch.tensor_split 4 in v1.8 without having to upgrade to 1.8 
predicted_ids = torch.tensor([100, 100, 31090, 100, 100, 100, 31090])
indices = torch.tensor([2,6])
np.array_split(predicted_ids,indices)
# returns [tensor([100, 100]), tensor([31090,   100,   100,   100]), tensor([31090])]

I would like the last line of the above, except via pytorch operations so I can avoid breaking the computational graph.
Is there a succinct way to do this?