mlx.core.logcumsumexp

logcumsumexp(a: array, /, axis: int | None = None, *, reverse: bool = False, inclusive: bool = True, stream: None | Stream | Device = None) → array

返回沿给定轴的元素的累积 logsumexp。

参数:

• a (array) – 输入数组

• axis (int, optional) – 可选的轴，用于计算累积 logsumexp。如果未指定，则返回展平数组的累积 logsumexp。

• reverse (bool) – 反向执行累积 logsumexp。

• inclusive (bool) – 输出的第 i 个元素包含输入的第 i 个元素。

返回值:

输出数组。

返回类型:

array