jax.numpy.modf

jax.numpy.modf(x, /, out=None)

傳回輸入陣列的元素級小數和整數部分。

JAX 版本的 numpy.modf。

參數:

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

• out – JAX 未使用。

傳回:

一個陣列，包含 x 元素的的小數和整數部分，提升 dtypes 為非精確型別。

傳回型別:

tuple[Array, Array]

另請參閱

• jax.numpy.divmod()：計算 x1 除以 x2 的元素級整數商和餘數。

範例

>>> jnp.modf(4.8)
(Array(0.8000002, dtype=float32, weak_type=True), Array(4., dtype=float32, weak_type=True))
>>> x = jnp.array([-3.4, -5.7, 0.6, 1.5, 2.3])
>>> jnp.modf(x)
(Array([-0.4000001 , -0.6999998 ,  0.6       ,  0.5       ,  0.29999995],      dtype=float32), Array([-3., -5.,  0.,  1.,  2.], dtype=float32))