softmax

Adds a Softmax activation along the given dimension.