fix bugs of softmax module, at the top layer of stacked AE