jax.scipy.special.xlog1py#
- jax.scipy.special.xlog1py = <jax._src.custom_derivatives.custom_jvp object>[源代码]#
计算 x*log(1 + y)，当 x=0 时返回 0。
scipy.special.xlog1py的 JAX 实现。定义为当\((x, y) = (0, -1)\)时返回 0，并使用自定义导数规则，以便在这种情况下自动微分是良好定义的。
- 参数:
x (ArrayLike) – 类数组，实数值。
y (ArrayLike) – 类数组，实值。
- 返回:
包含 xlog1py 值的数组。
- 返回类型: