max_v= torch.max(scores_s, 1)[0]

下面的代码是一样的：

  index_num = torch.arange(0, conf.size(0))
                    max_ids= torch.max(scores_s, 1)[1]

                    scores_v=scores_s[index_num,max_ids]

筛选满足条件的行或列（使用与或）

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      x = torch.linspace(1, 8, steps=8).view(4, 2)
     
    
   

    

     

    
    

     

      print(x)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      area1=(x[:,0]>5.5)&(x[:,1]>5.5)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      c=x[:,0]*x[:,1]
     
    
   

    

     

    
    

     

      area2=c>25
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      area=area1|area2
     
    
   

    

     

    
    

     

      print(x[area])
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      if 0:
     
    
   

    

     

    
    

     

      # index=torch.max(area,1)[0]
     
    
   

    

     

    
    

     

      b=x[area]
     
    
   

    

     

    
    

     

      # b= x[torch.where((x[:,0]>0) & (x[:,0]<6))]
     
    
   

    

     

    
    

     

      # print(b)
     
    
  
 

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

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