jax.numpy.logaddexp2

jax.numpy.logaddexp2 = <jnp.ufunc 'logaddexp2'>

输入在底数为2的指数和的对数，避免溢出。

NumPy 中 numpy.logaddexp2 的 JAX 实现。

参数:

• x1 – 输入数组或标量。

• x2 – 输入数组或标量。x1 和 x2 的形状应相同或可广播兼容。

• args (ArrayLike)

• out (None)

• where (None)

返回:

一个数组，其中包含逐元素的计算结果 \(log_2(2^{x1}+2^{x2})\)。

返回类型:

任意类型

另请参阅

• jax.numpy.logaddexp(): 逐元素计算 log(exp(x1) + exp(x2))。

• jax.numpy.log2(): 逐元素计算 x 的底数为2的对数。

示例

>>> x1 = jnp.array([[3, -1, 4],
...                 [8, 5, -2]])
>>> x2 = jnp.array([2, 3, -5])
>>> result1 = jnp.logaddexp2(x1, x2)
>>> result2 = jnp.log2(jnp.exp2(x1) + jnp.exp2(x2))
>>> jnp.allclose(result1, result2)
Array(True, dtype=bool)