jax.numpy.log#
- jax.numpy.log(x, /)[原始碼]#
計算輸入的元素級自然對數。
JAX 版本的
numpy.log
。- 參數:
x (ArrayLike) – 輸入陣列或純量。
- 返回:
一個陣列,包含
x
中每個元素的對數,提升為非精確 dtype。- 返回類型:
參見
jax.numpy.exp()
:計算輸入的元素級指數。jax.numpy.log2()
:計算輸入中每個元素的以 2 為底的對數。jax.numpy.log1p()
:計算輸入加一後的元素級對數。
範例
jnp.log
和jnp.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)