jax.scipy.special.kl_div

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

The Kullback-Leibler divergence.

JAX implementation of 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}\]

Parameters:

• p (ArrayLike) – arraylike, real-valued.

• q (ArrayLike) – arraylike, real-valued.

Returns:

array of KL-divergence values

Return type:

Array

See also

• jax.scipy.special.entr()

• jax.scipy.special.rel_entr()