Я новичок в PyTorch. Мне нужно сохранить идентификатор для моего набора проверки в GPU. Например, у меня будет один идентификатор, скажем 1012, теперь мне нужно создать другой набор, например:
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) ]
)
...
Я хочу, чтобы у меня была тензорная переменная в графическом процессоре, например:
tensor [1012][results1]
tensor [1013][results2]
А также, как получить для каждого индекса. Любая помощь будет оценена по достоинству.