jax.numpy.logaddexp2

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

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

numpy.logaddexp2 的 JAX 实现。

参数:

• x1 – 输入数组或标量。

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

• args (ArrayLike)

• out (None)

• where (None)

返回:

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

返回类型:

Any

另请参阅

• 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)