jax.numpy.log#

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

計算輸入的元素級自然對數。

JAX 版本的 numpy.log

參數:

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

返回:

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

返回類型:

Array

參見

範例

jnp.logjnp.exp 是彼此的反函數。將 jnp.log 應用於 jnp.exp(x) 的結果會產生原始輸入 x

>>> x = jnp.array([2, 3, 4, 5])
>>> jnp.log(jnp.exp(x))
Array([2., 3., 4., 5.], dtype=float32)

使用 jnp.log 我們可以示範眾所周知的對數性質,例如 \(log(a*b) = log(a)+log(b)\)

>>> x1 = jnp.array([2, 1, 3, 1])
>>> x2 = jnp.array([1, 3, 2, 4])
>>> jnp.allclose(jnp.log(x1*x2), jnp.log(x1)+jnp.log(x2))
Array(True, dtype=bool)