torch.sum

exp指数函数，把所有的y都整到0以上，不用担心-0.5和0.5抵消的问题

  

   

    

     

    
    

     
 import torch
     
    
   

    

     

    
    

     
 import numpy as np
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       data=np.array([[[0.5,-0.5],[-0.05,-0.05]]])
     
    
   

    

     

    
    

     

       x = torch.from_numpy(data.astype(np.float32))
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       aaa=torch.exp(torch.Tensor([1]))
     
    
   

    

     

    
    

     

       print(aaa)
     
    
   

    

     

    
    

     

       print(x)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       print(x.sum(2))  # 按行求和
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

       x=torch.exp(x)
     
    
   

    

     

    
    

     

       print(x.sum(2))  # 按行求和
     
    
   

    

     

    
    

     

       print(x.sum(2)[x.sum(2)>aaa])  # 按行求和
     
    
  
 

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

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