add a batch version of softmax