jax.scipy.special.logit

Warning

This page was created from a pull request (#9655).

jax.scipy.special.logit¶

jax.scipy.special.logit = <jax._src.custom_derivatives.custom_jvp object>[source]¶

Logit ufunc for ndarrays.

LAX-backend implementation of logit().

Original docstring below.

The logit function is defined as logit(p) = log(p/(1-p)). Note that logit(0) = -inf, logit(1) = inf, and logit(p) for p<0 or p>1 yields nan.

Parameters

x (ndarray) – The ndarray to apply logit to element-wise.

Returns

out – An ndarray of the same shape as x. Its entries are logit of the corresponding entry of x.

Return type

ndarray