Given two tensors:
A = torch.tensor([1,2,3]) B = torch.tensor([3,5])
Then, I want to obtain the tensor C = tensor([1,2]), in whi
C = tensor([1,2])