jax.scipy.special.xlog1py

内容

jax.scipy.special.xlog1py#

jax.scipy.special.xlog1py = <jax._src.custom_derivatives.custom_jvp object>[source]#

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

JAX 实现 scipy.special.xlog1py

\((x, y) = (0, -1)\) 时,此函数定义为返回 0,并使用自定义导数规则,以便在此点自动微分定义良好。

参数:
  • x (ArrayLike) – 数组类,实值。

  • y (ArrayLike) – 数组类,实值。

返回值:

包含 xlog1py 值的数组。

返回类型:

数组