improve the  performance of softmax