jax.numpy.exp2

jax.numpy.exp2(x, /)

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

JAX 实现的 numpy.exp2。

参数:

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

返回:

一个数组，包含 x 中每个元素的以 2 为底的指数，并提升为非精确 dtype。

返回类型:

数组

另请参阅

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