Fix bug in log_softmax kernel when lastdim is larger than 100000

There is an unexpected log in the calculation

Cherry-Pick: #53654