torch.norm

范数1，绝对值求和

范数2，平方和求开根号2

范数3，平方和求开根号3

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.tensor as tensor
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      a = tensor([[-1, 2, 3, 4],
     
    
   

    

     

    
    

     

       [1, 2, 3, 4]]).float()  # norm仅支持floatTensor,a是一个2*4的Tensor
     
    
   

    

     

    
    

     

      a0 = torch.norm(a, p=3, dim=0)  # 按0维度求2范数
     
    
   

    

     

    
    

     

      a1 = torch.norm(a, p=3, dim=1)  # 按1维度求2范数
     
    
   

    

     

    
    

     

      print(a0)
     
    
   

    

     

    
    

     

      print(a1)
     
    
  
 

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

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