_fftlog_multimethods.py 575 B

1234567891011121314151617181920212223242526272829
  1. '''Multimethods for fast Hankel transforms.
  2. '''
  3. import numpy as np
  4. from ._basic import _dispatch
  5. from ._fftlog import fht as _fht
  6. from ._fftlog import ifht as _ifht
  7. from scipy._lib.uarray import Dispatchable
  8. __all__ = ['fht', 'ifht']
  9. @_dispatch
  10. def fht(a, dln, mu, offset=0.0, bias=0.0):
  11. """fht multimethod."""
  12. return (Dispatchable(a, np.ndarray),)
  13. @_dispatch
  14. def ifht(A, dln, mu, offset=0.0, bias=0.0):
  15. """ifht multimethod."""
  16. return (Dispatchable(A, np.ndarray),)
  17. # copy over the docstrings
  18. fht.__doc__ = _fht.__doc__
  19. ifht.__doc__ = _ifht.__doc__