mlx.core.logaddexp

目录

mlx.core.logaddexp#

logaddexp(a: scalar | array, b: scalar | array, /, *, stream: None | Stream | Device = None) array#

逐元素的对数加指数运算。

这是两个数组的数值稳定版对数加指数运算,具有 NumPy 风格的广播语义。输入数组可以是一个或两个,也可以是标量。

该计算是 log(exp(a) + exp(b)) 的数值稳定版本。

参数:
  • a (array) – 输入数组或标量。

  • b (array) – 输入数组或标量。

返回值:

ab 的对数加指数运算结果。

返回类型:

array