jax.scipy.special.xlog1py#

jax.scipy.special.xlog1py = <jax._src.custom_derivatives.custom_jvp object>[源代码]#

计算 x*log(1 + y)，当 x=0 时返回 0。

JAX 实现的 scipy.special.xlog1py。

定义当 \((x, y) = (0, -1)\) 时返回 0，并具有自定义导数规则，以便在此点自动微分得到良好定义。

参数:

  • x (类数组) – 类数组，实数值。

  • y (类数组) – 类数组，实数值。

返回:

包含 xlog1py 值的数组。

返回类型:

数组

另请参阅

jax.scipy.special.xlogy()