jax.numpy.log

jax.numpy.log(x, /)

计算输入的逐元素自然对数。

JAX 对 numpy.log 的实现。

参数:

x (ArrayLike) – 输入数组或标量。

返回:

一个包含 x 中每个元素的对数的数组，会提升为非精确的 dtype。

返回类型:

Array

另请参阅

• jax.numpy.exp()：计算输入的逐元素指数。

• jax.numpy.log2()：计算输入的每个元素的以2为底的对数。

• jax.numpy.log1p()：计算输入值加 1 的逐元素对数。

示例

jnp.log 和 jnp.exp 互为反函数。将 jnp.exp(x) 的结果应用于 jnp.log 会得到原始输入 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)