pytorch argsort

维度1上排序，并且倒叙输出。

index_score=score_mat.argsort(axis=1).flip(dims=[1])
 

index_score=score_mat.argsort(axis=1,descending=True)
 

经过测试，效果一样的：

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      data=torch.randn(4,4)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(data)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      index_score=data.argsort(axis=1).flip(dims=[1])
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(index_score)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      index_score2=data.argsort(axis=1,descending=True)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(index_score2)
     
    
  
 

pytorch反转，不是倒叙：

  

   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      data=torch.arange(10).view(2, 5)
     
    
   

    

     

    
    

     

      print(data)
     
    
   

    

     

    
    

     

      aaa=torch.flip(data, dims=[ 1])
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(aaa)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      bbb=torch.flip(data, dims=[0])	# 对第0维进行反转
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(bbb)
     
    
   

    

     

    
    

     

      ccc= torch.flip(data, dims=[0, 1])	# 对第0、1维进行反转
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(ccc)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      ddd= data.flip(dims=[0, 1])	# 对第0、1维进行反转，与上一句效果相同
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(ddd)
     
    
  
 

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/119707847