jax.numpy.exp2

jax.numpy.exp2(x, /)

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

JAX 对 numpy.exp2 的实现。

参数:

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

返回:

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

返回类型:

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)