Make a ragged tensor from a normal tensor

Question

If I have a tensor like this one

[[[1, 2], [2, 6], [7, 8]],
 [[7, 9], [1, 6], [3, 2]],
 [[8, 3], [3, 6], [3, 9]]]

and another tensor which specif