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)