torch.Tensorからアイテムを削除

Question

私は次のコードをwritainに持っています。

N最大スコアのインデックスをscoresから取得したいと思います。

現在の最大値をscoresから繰り返し削除して最大値を再度取得する必要がありますが、それを行う適切な方法が見つからないようです。

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

あなたが表示されたコード（特にfoorループが）本当にあなたがやりたいたいに関連していないようですので、私は、私が誤解していなかった願っています。とにかく、ここで私はそれをやるでしょう。

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)) 

希望します。