jax.numpy.exp2

jax.numpy.exp2(x, /)

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

JAX 实现 numpy.exp2。

参数：

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

返回值：

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

返回类型：

数组

参见

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