jax.numpy.ldexp

jax.numpy.ldexp(x1, x2, /)

計算 x1 * 2 ** x2

JAX 實作的 numpy.ldexp()。

請注意，XLA 沒有提供 ldexp 運算，因此這在 JAX 中透過標準乘法和指數運算實作。

參數:

• x1 (ArrayLike) – 實數值輸入陣列。

• x2 (ArrayLike) – 整數輸入陣列。必須與 x1 廣播相容。

傳回:

逐元素計算的 x1 * 2 ** x2。

傳回類型:

Array

參見

• jax.numpy.frexp()：將值分解為假數和指數。

範例

>>> x1 = jnp.arange(5.0)
>>> x2 = 10
>>> jnp.ldexp(x1, x2)
Array([   0., 1024., 2048., 3072., 4096.], dtype=float32)

ldexp 可用於重建 frexp 的輸入

>>> x = jnp.array([2., 3., 5., 11.])
>>> m, e = jnp.frexp(x)
>>> m
Array([0.5   , 0.75  , 0.625 , 0.6875], dtype=float32)
>>> e
Array([2, 2, 3, 4], dtype=int32)
>>> jnp.ldexp(m, e)
Array([ 2.,  3.,  5., 11.], dtype=float32)