Softmax implemeation is also OK!

Signed-off-by: Zhou Chang <achang.zhou@gmail.com>