jax.scipy.special.kl_div#

jax.scipy.special.kl_div(p, q)[源代码]#

Kullback-Leibler 散度。

scipy.special.kl_div 的 JAX 实现。

$$\begin{array}{r}\mathrm{kl}\mathrm{\_}\mathrm{div}(p,q)=\{\begin{array}{ll}p\log (p/q)-p+q& p>0,q>0\\ q& p=0,q\ge 0\\ \mathrm{\infty }& \mathrm{otherwise}\end{array}\end{array}$$

参数：

• p (ArrayLike) – 类数组，实数值。

• q (ArrayLike) – 类数组，实数值。

返回：

KL 散度值的数组

返回类型：

Array

另请参阅

• jax.scipy.special.entr()

• jax.scipy.special.rel_entr()