[MRG] softmax function that builds expression instead of using softmax op