Softmax loss function. .