jax.numpy.exp2

jax.numpy.exp2(x, /)

计算输入的逐元素底数为2的指数。

JAX 对 numpy.exp2 的实现。

参数:

x (ArrayLike) – 输入数组或标量

返回:

一个数组，其中包含 x 中每个元素的底数为2的指数，提升为不精确的数据类型。

返回类型:

Array

另请参阅

• jax.numpy.log2(): 计算输入的每个元素的底数为2的对数。

• jax.numpy.exp(): 计算输入的每个元素的指数。

• jax.numpy.expm1(): 计算输入的每个元素的 \(e^x-1\)。

示例

jnp.exp2 遵循指数的性质，例如 \(2^{a+b} = 2^a * 2^b\)。

>>> x1 = jnp.array([2, -4, 3, -1])
>>> x2 = jnp.array([-1, 3, -2, 3])
>>> jnp.exp2(x1+x2)
Array([2. , 0.5, 2. , 4. ], dtype=float32)
>>> jnp.exp2(x1)*jnp.exp2(x2)
Array([2. , 0.5, 2. , 4. ], dtype=float32)