jax.scipy.special.kl_div

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

库尔贝克-莱布勒散度。

JAX implementation of scipy.special.kl_div.

\[ \begin{align}\begin{aligned}\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}\\\begin{split}KL散度(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}\end{aligned}\end{align} \]

参数:

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

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

返回:

KL-散度值的数组

返回类型:

Array

参见

• jax.scipy.special.entr()

• jax.scipy.special.rel_entr()