softmax is badly named and should rather be called softargmax.