I am trying to implement one Keras code in Pytorch. I am little confused with the loss function in Keras and Pytroch. The loss function used in Keras model is categorical cross entropy but I haven’t heard of categorical cross entropy in Pytorch. I am using MNIST dataset. I would like to know how to implement categorical cross entropy in Pytorch.