fix sofmax cross entropy loss and its test