Function logsumexp

Log-sum-exp reduction. Also a multivariate version of softplus.

If no axis is specified, the reduction is performed over all elements. This convention differs from jax.nn.logSoftmax().

Reference: https://en.wikipedia.org/wiki/LogSumExp