Function logsumexp_keepdims 

pub fn logsumexp_keepdims(a: &Array, axes: impl Axes) -> Array

log(sum(exp(a))), computed stably, over axes, keeping them as size-1 dimensions; see Array::logsumexp_keepdims.