Medium – 11 Oct 18
  
  
    

Understanding Cross Entropy implementation in Pytorch (softmax, log_softmax,... 6

This notebook breaks down how `cross_entropy` function is implemented in pytorch, and how it is related to softmax, log_softmax, and nll…

  Reading time: 2 min read
    

  
  
    
    
  
  


The inclusion of softmax is because of taking numerical stability on account.