给定一个张量：
A = torch.tensor([2., 3., 4., 5., 6., 7.])
然后，给每个元素A一个 id：
id = torch.arange(A.shape[0], dtype = torch.int) # tensor([0,1,2,3,4,5])
换句话说，2.in的idA为0，3.in的idA为1：
2. -> 0
3. -> 1
4. -> 2
5. -> 3
6. -> 4
7. -> 5
然后，我有一个新的张量：
B = torch.tensor([3., 6., 6., 5., 4., 4., 4.])
在pytorch中，Pytorch中有什么方法可以将B中的每个元素映射到id吗？换句话说，我想获取tensor([1, 4, 4, 3, 2, 2, 2])，其中每个元素都是 中元素的 id B。