negative log loss and cross entropy in pytorch