fix test softmax with loss layer