Fixed logsumexp with axis.