jax.numpy.exp2

jax.numpy.exp2(x, /)

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

JAX implementation of 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)