jax.scipy.special.kl_div

jax.scipy.special.kl_div(p, q)

Kullback-Leibler 散度。

JAX 实现 scipy.special.kl_div。

\[\begin{split} \mathrm{kl\_div}(p, q) = \begin{cases} p\log(p/q)-p+q & p>0,q>0\\ q & p=0,q\ge 0\\ \infty & \mathrm{otherwise} \end{cases}\end{split}\]

参数:

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

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

返回:

KL 散度值的数组

返回类型:

数组

另请参阅

• jax.scipy.special.entr()

• jax.scipy.special.rel_entr()