TensorPlay
Reference guides
Copy
View MarkdownDownload .md

tensorplay.special

Functions

tensorplay.functional.airy_ai

tensorplay.functional.bessel_j0

tensorplay.functional.bessel_j1

tensorplay.functional.bessel_y0

tensorplay.functional.bessel_y1

tensorplay.functional.chebyshev_polynomial_t

tensorplay.functional.chebyshev_polynomial_u

tensorplay.functional.chebyshev_polynomial_v

tensorplay.functional.chebyshev_polynomial_w

tensorplay.functional.digamma

tensorplay.special.entr

Entropy of a probability element: -x*log(x) (0 at x=0, NaN below).

tensorplay.functional.erf

tensorplay.functional.erfc

tensorplay.special.erfcx

Scaled complementary error function: exp(x^2) * erfc(x).

tensorplay.functional.erfinv

tensorplay.functional.exp2

tensorplay.special.expit

Numerically-stable sigmoid; identical to sigmoid().

tensorplay.functional.expm1

tensorplay.functional.gammainc

tensorplay.functional.gammaincc

tensorplay.special.gammaln

Natural log of the absolute value of the gamma function (native lgamma).

tensorplay.functional.hermite_polynomial_h

tensorplay.functional.hermite_polynomial_he

tensorplay.functional.i0

tensorplay.functional.i0e

tensorplay.functional.i1

tensorplay.functional.i1e

tensorplay.functional.laguerre_polynomial_l

tensorplay.functional.legendre_polynomial_p

tensorplay.functional.log1p

tensorplay.special.log_ndtr

log of the standard normal CDF, stable in the left tail.

tensorplay.special.log_softmax

tensorplay.functional.logit

tensorplay.special.logsumexp

Log-sum-exp along dim, computed with the max-shift trick.

tensorplay.special.modified_bessel_i0

Modified Bessel function of the first kind, order 0; alias of i0().

tensorplay.functional.modified_bessel_i1

tensorplay.functional.modified_bessel_k0

tensorplay.functional.modified_bessel_k1

tensorplay.special.multigammaln

Multivariate log-gamma with dimension p (p >= 2).

tensorplay.special.ndtr

Standard normal CDF: Phi(x) = 0.5 * erfc(-x / sqrt(2)).

tensorplay.special.ndtri

Inverse standard normal CDF: sqrt(2) * erfinv(2x - 1).

tensorplay.special.polygamma

Polygamma of order :attr:n: \(\psi^{(n)}(x)\) (native kernel).

tensorplay.special.psi

Digamma function; alias of digamma().

tensorplay.special.round

Rounds to nearest even integer (native Tensor.round).

tensorplay.functional.scaled_modified_bessel_k0

tensorplay.functional.scaled_modified_bessel_k1

tensorplay.functional.shifted_chebyshev_polynomial_t

tensorplay.functional.shifted_chebyshev_polynomial_u

tensorplay.functional.shifted_chebyshev_polynomial_v

tensorplay.functional.shifted_chebyshev_polynomial_w

tensorplay.functional.sinc

tensorplay.special.softmax

tensorplay.functional.spherical_bessel_j0

tensorplay.special.xlog1py

x*log1p(y) with the convention x*log1p(-1) = 0 when x == 0.

tensorplay.special.xlogy

x*log(y) with the convention 0*log(anything) = 0.

tensorplay.functional.zeta

Ask DeepWiki