mlx.core.logsumexp

logsumexp(a: array, /, axis: None | int | Sequence[int] = None, keepdims: bool = False, *, stream: None | Stream | Device = None) → array

对给定轴执行 log-sum-exp 约简。

log-sum-exp 约简是以下操作在数值上更稳定的版本：

log(sum(exp(a), axis))

参数:

• a (array) – 输入数组。

• axis (int 或 list(int), 可选) – 可选的轴或多个轴，用于进行约简。如果未指定，默认为对整个数组进行约简。

• keepdims (bool, 可选) – 保持被约简的轴作为单维度，默认为 False。

返回:

输出数组，对应轴已约简。

返回类型:

array