jax.numpy.frexp

jax.numpy.frexp(x, /)

將浮點數值分割成尾數和二的指數。

JAX 版本的 numpy.frexp() 實作。

參數:

x (ArrayLike) – 實數值陣列

回傳:

一個元組 (mantissa, exponent)，其中 mantissa 是一個介於 -1 和 1 之間的浮點數值，而 exponent 是一個整數，使得 x == mantissa * 2 ** exponent。

回傳類型:

tuple[Array, Array]

另請參閱

• jax.numpy.ldexp()：計算 frexp 的反函數。

範例

將數值分割成尾數和指數

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

重建原始陣列

>>> m * 2 ** e
Array([1., 2., 3., 4., 5.], dtype=float32)