Pytorch取最小或最大的张量索引

Pytorch中根据索引取张量有很多方法，比如index_select和masked_select，和gt,ge等配合食用，但如果需要取出最小几个或最大几个张量的索引，则需要动手写一下 a = torch.tensor([2,3,1,5]) y,_ = torch.sort(a) mask = a.