mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
d6b030856b
Adds prims and refs for special functions (bessel_j0, bessel_j1, spherical_bessel_j0). Thanks! Pull Request resolved: https://github.com/pytorch/pytorch/pull/86049 Approved by: https://github.com/mruberry
53 lines
1.3 KiB
ReStructuredText
53 lines
1.3 KiB
ReStructuredText
.. role:: hidden
|
|
:class: hidden-section
|
|
|
|
torch.special
|
|
=============
|
|
|
|
The torch.special module, modeled after SciPy's `special <https://docs.scipy.org/doc/scipy/reference/special.html>`_ module.
|
|
|
|
.. automodule:: torch.special
|
|
.. currentmodule:: torch.special
|
|
|
|
Functions
|
|
-----------------------
|
|
|
|
.. autofunction:: airy_ai
|
|
.. autofunction:: bessel_j0
|
|
.. autofunction:: bessel_j1
|
|
.. autofunction:: digamma
|
|
.. autofunction:: entr
|
|
.. autofunction:: erf
|
|
.. autofunction:: erfc
|
|
.. autofunction:: erfcx
|
|
.. autofunction:: erfinv
|
|
.. autofunction:: exp2
|
|
.. autofunction:: expit
|
|
.. autofunction:: expm1
|
|
.. autofunction:: gammainc
|
|
.. autofunction:: gammaincc
|
|
.. autofunction:: gammaln
|
|
.. autofunction:: i0
|
|
.. autofunction:: i0e
|
|
.. autofunction:: i1
|
|
.. autofunction:: i1e
|
|
.. autofunction:: log1p
|
|
.. autofunction:: log_ndtr
|
|
.. autofunction:: log_softmax
|
|
.. autofunction:: logit
|
|
.. autofunction:: logsumexp
|
|
.. autofunction:: multigammaln
|
|
.. autofunction:: ndtr
|
|
.. autofunction:: ndtri
|
|
.. autofunction:: polygamma
|
|
.. autofunction:: psi
|
|
.. autofunction:: round
|
|
.. autofunction:: scaled_modified_bessel_k0
|
|
.. autofunction:: scaled_modified_bessel_k1
|
|
.. autofunction:: sinc
|
|
.. autofunction:: softmax
|
|
.. autofunction:: spherical_bessel_j0
|
|
.. autofunction:: xlog1py
|
|
.. autofunction:: xlogy
|
|
.. autofunction:: zeta
|