fix softmax cross entropy loss layer