如何创建具有多个索引的张量索引？

Question

我是PyTorch的新手。我需要在GPU中存储我的验证集ID。例如，我将有一个ID表示1012，现在我需要创建另一个ID，例如：

tensor results1=(
[102(another id), 12(rank) ], //(another id) and (rank) will not be there
[103(another id),  1(rank) ],
[104(another id),  5(rank) ],
[104(another id) , 1(rank) ]
) 
...

我想要的是我可以在GPU中拥有一个张量变量：

tensor [1012][results1]
tensor [1013][results2]

以及如何检索每个索引。 任何帮助将不胜感激。