New softmax definition