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.
This should work: