Supprimer l'article de torch.Tensor

Question

J'ai le code suivant writtein dans lua.

Je voudrais obtenir des indices pour N scores maximum de scores et leurs scores correspondants.

Il semble que je vais devoir itérativement supprimer la valeur maximale actuelle de scores et récupérer le maximum à nouveau, mais ne trouve pas un moyen approprié de le faire.

nqs=dataset['question']:size(1); 
scores=torch.Tensor(nqs,noutput); 
qids=torch.LongTensor(nqs); 
for i=1,nqs,batch_size do 
    xlua.progress(i, nqs) 
    r=math.min(i+batch_size-1,nqs); 
    scores[{{i,r},{}}],qids[{{i,r}}]=forward(i,r); 
-- print(scores) 
end 

tmp,pred=torch.max(scores,2); 

Answer 1

J'espère que je ne l'ai pas mal compris, puisque le code que vous montrer (en particulier la boucle Foor) ne semble pas vraiment pertinent de vouloir que vous voulez faire. Quoi qu'il en soit, voici comment je le ferais.

sr=scores:view(-1,scores:size(1)*scores:size(2)) 
val,id=sr:sort() 
--val is a row vector with the values stored in increasing order 
--id will be the corresponding index in sr 
--now you can slice val and id from the end to find the N values you want, then you can recover the original index in the scores matrix simply with 
col=(index-1)%scores:size(2)+1 
row=math.ceil(index/scores:size(2)) 

espérons que cette aide.