pytorch CrossEntropyLoss测试

测试代码：

  

   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      loss_fn = torch.nn.CrossEntropyLoss(reduce=True, weight=None)
     
    
   

    

     

    
    

     

      input = torch.Tensor([0,0,0,0.5,0.6,.7])
     
    
   

    

     

    
    

     

      target = torch.LongTensor(3).random_(5)
     
    
   

    

     

    
    

     

      print(input)
     
    
   

    

     

    
    

     

      print(target)
     
    
   

    

     

    
    

     

      loss = loss_fn(input, target)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      print(loss)
     
    
  
 

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

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