Factor softmax out to a helper function.

Add the softmax output to the activation factory as well.