mlx.core.logaddexp

Contents

mlx.core.logaddexp#

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

Element-wise log-add-exp.

This is a numerically stable log-add-exp of two arrays with numpy-style broadcasting semantics. Either or both input arrays can also be scalars.

The computation is is a numerically stable version of log(exp(a) + exp(b)).

Parameters:
  • a (array) – Input array or scalar.

  • b (array) – Input array or scalar.

Returns:

The log-add-exp of a and b.

Return type:

array