softmax python version updates, using vectorization.
Vectorize softmax module.