jax.numpy.expm1#

jax.numpy.expm1(x, /)[原始碼]#

計算輸入中每個元素的 exp(x)-1

JAX 實現的 numpy.expm1

參數:

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

返回值:

包含 x 中每個元素的 exp(x)-1 的陣列,會提升為非精確資料型別。

返回型別:

Array

注意

對於 `x` 的小數值,jnp.expm1exp(x)-1 的簡單計算具有更高的精度。

另請參閱

範例

>>> x = jnp.array([2, -4, 3, -1])
>>> with jnp.printoptions(precision=2, suppress=True):
...   print(jnp.expm1(x))
[ 6.39 -0.98 19.09 -0.63]
>>> with jnp.printoptions(precision=2, suppress=True):
...   print(jnp.exp(x)-1)
[ 6.39 -0.98 19.09 -0.63]

對於非常接近 0 的值,jnp.expm1(x)jnp.exp(x)-1 更準確

>>> x1 = jnp.array([1e-4, 1e-6, 2e-10])
>>> jnp.expm1(x1)
Array([1.0000500e-04, 1.0000005e-06, 2.0000000e-10], dtype=float32)
>>> jnp.exp(x1)-1
Array([1.00016594e-04, 9.53674316e-07, 0.00000000e+00], dtype=float32)