I have two tensors like this:
a = tensor([1,3,4,5]).cuda() b = tensor([0,1,2,3,4,5,6]).cuda()
I want the result tensor to be like this:<