pub fn logsumexp(a: &Array, axes: &[isize], keepdims: bool) -> Result<Array>Expand description
log(sum(exp(a))) over axes (all axes if empty), computed stably as MLX does: the
maximum is subtracted first (without gradient), and rows whose maximum is infinite
return that maximum.