Add comments to _multinomial_loss_grad_hess function.