pytorch 随机打乱

参考：https://stackoverflow.com/questions/44738273/torch-how-to-shuffle-a-tensor-by-its-rows

一维的：

  

   

    

     

    
    

     

      x = torch.arange(0, 10)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(x)
     
    
   

    

     

    
    

     

      b=torch.randperm(x.size(0))
     
    
   

    

     

    
    

     

      print(x[b])
     
    
  
 

  

   

    

     

    
    

     

      a=torch.rand(3,5)
     
    
   

    

     

    
    

     

      print(a)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      a=a[torch.randperm(a.size(0))]
     
    
   

    

     

    
    

     

      print(a)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      a=a[:,torch.randperm(a.size(1))]
     
    
   

    

     

    
    

     

      print(a)
     
    
  
 

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

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