Categorical cross entropy loss function equivalent in PyTorch

@ptrblck, I want something like below image.


1 Like