jax.nn.log1mexp

jax.nn.log1mexp = <jax._src.custom_derivatives.custom_jvp 对象>

数值稳定地计算 \(\log(1 - \exp(-x))\)。

此函数在 \(x < 0\) 时未定义。

基于 TensorFlow 的实现。

参考文献

[1]

Martin Mächler. Accurately Computing log(1 − exp(−|a|)) Assessed by the Rmpfr package.（准确计算 log(1 − exp(−|a|))，经 Rmpfr 包评估）。

参数:

x (ArrayLike)

返回类型:

Array