jax.lax.log1p

jax.lax.log1p(x, accuracy=None)

逐元素计算 \(\mathrm{log}(1 + x)\)。

此函数直接降低到 stablehlo.log_plus_one 操作。 与朴素表达式 lax.log(1 + x) 相比，它对于接近零的 x 更准确。

参数:

• x (ArrayLike) – 输入数组。 必须具有浮点型或复数类型。

• accuracy – 可选的 lax.Tolerance 或 lax.AccuracyMode 对象，用于根据请求的精度选择 op 的实现。如果实现无法满足请求的容差，编译器将返回错误。如果指定了 mode 并且没有多个实现可用，则将使用默认实现。

返回:

包含 x 的逐元素自然对数 x + 1 的相同形状和 dtype 的数组。

返回类型:

数组

另请参阅

• jax.lax.expm1(): 逐元素计算 \(e^x - 1\)。

• jax.lax.log(): 逐元素自然对数 \(\mathrm{log}(x)\)。