How to convert topk data from torch.tensor.topk() to a pytorch tensor

Question

I have a 1-d pytorch tensor and I got topk data of the tensor and indeces of this datas. How can I place each data in the corresponding position of a empty pytorch tensor?

The topk of this tensor elements and their indeces is got by: value, indeces = toech.topk(T,K) and I also have an empty tensor t of the same size as T. How can I copy each value[i] to t[indeces[i]]? I have tried to use for i in value but this way is too slow when data isn't very small.

Can't you just do `t[indeces] = value` ? – Toyo Apr 19 '23 at 04:04 — Toyo, Apr 19 '23 at 04:04
@Toyo `t[indeces] = value` works! Thanks! – CXLi Apr 19 '23 at 06:08 — CXLi, Apr 19 '23 at 06:08

score 0 · Accepted Answer · answered Apr 19 '23 at 06:10

0

This should work:

t[indeces] = value

answered Apr 19 '23 at 06:10

Toyo

667
1
5
22

How to convert topk data from torch.tensor.topk() to a pytorch tensor

1 Answers1