Pytorch最小または最大インデックステンソルを取ります

インデックス内のPytorchテイクテンソルあり食品となどindex_selectとmasked_select、およびGT、GE、など多くの方法は、ですが、あなたが最小または複数のテンソル指数の最大数を削除する必要がある場合、あなたは手について記述する必要があります

A = torch.tensor（[2,3,1,5 ]）
Y、_ = torch.sort（）
マスク = a.gt（Y [0]）
インデックス = [] 
mask_list =（マスク== 偽）。ゼロ以外の（）
のインデックス = [INT（I）のために私にmask_list] 
インデックス

 >>> [2]

マスクの最初のソートテンソルを行い、元のシートを得ることは最小値よりも大きく、以下で、Y [0]が真であるより大きい未満、またはFalseに等しい、次いでnonezero（）メソッドは、マスク偽を得るために使用することができます行われたインデックス、