jax.scipy.special.entr

Contents

jax.scipy.special.entr#

jax.scipy.special.entr(x)[source]#

The entropy function

JAX implementation of scipy.special.entr.

\[\begin{split}\mathrm{entr}(x) = \begin{cases} -x\log(x) & x > 0 \\ 0 & x = 0\\ -\infty & x > 0 \end{cases}\end{split}\]
Parameters:

x (jax.typing.ArrayLike) – arraylike, real-valued.

Returns:

array containing entropy values.

Return type:

Array