Hello,
Does anyone know how to split batches of tensors in different positions each?
I am trying to separate my tensors in two parts according to the split_position, which is different for each sample
batch.shape = torch.Size([128, 68, 1])      # (batch_size, max_len, 1)
split_positions.shape = ([128])             # split position of each sample

Thanks,