jax.numpy.log1p

jax.numpy.log1p(x, /)

計算輸入值加一後的元素級對數，log(x+1)。

JAX 實作的 numpy.log1p。

參數:

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

返回:

一個陣列，包含 x 中每個元素加一後的對數，並提升為非精確 dtype。

返回類型:

Array

注意

對於小的 x 值，jnp.log1p 比使用 log(x+1) 的樸素計算更準確。

參見

• jax.numpy.expm1()：計算輸入中每個元素的 \(e^x-1\)。

• jax.numpy.log2()：計算輸入中每個元素的以 2 為底的對數。

• jax.numpy.log()：計算輸入的元素級對數。

範例

>>> x = jnp.array([2, 5, 9, 4])
>>> jnp.allclose(jnp.log1p(x), jnp.log(x+1))
Array(True, dtype=bool)

對於非常接近 0 的值，jnp.log1p(x) 比 jnp.log(x+1) 更準確

>>> x1 = jnp.array([1e-4, 1e-6, 2e-10])
>>> jnp.expm1(jnp.log1p(x1))  
Array([1.00000005e-04, 9.99999997e-07, 2.00000003e-10], dtype=float32)
>>> jnp.expm1(jnp.log(x1+1))  
Array([1.000166e-04, 9.536743e-07, 0.000000e+00], dtype=float32)