jax.numpy.exp2

jax.numpy.exp2(x, /)

計算輸入的逐元素 2 的指數。

numpy.exp2 的 JAX 實作。

參數：

x (ArrayLike) – 輸入陣列或純量

回傳：

一個陣列，包含 x 中每個元素的 2 的指數，並提升為非精確 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)