1  from collections.abc import Sequence
  2  from typing import Literal as L
  3  
  4  from numpy import complex128, float64
  5  from numpy._typing import ArrayLike, NDArray, _ArrayLikeNumber_co
  6  
  7  _NormKind = L[None, "backward", "ortho", "forward"]
  8  
  9  __all__: list[str]
 10  
 11  def fft(
 12      a: ArrayLike,
 13      n: None | int = ...,
 14      axis: int = ...,
 15      norm: _NormKind = ...,
 16  ) -> NDArray[complex128]: ...
 17  
 18  def ifft(
 19      a: ArrayLike,
 20      n: None | int = ...,
 21      axis: int = ...,
 22      norm: _NormKind = ...,
 23  ) -> NDArray[complex128]: ...
 24  
 25  def rfft(
 26      a: ArrayLike,
 27      n: None | int = ...,
 28      axis: int = ...,
 29      norm: _NormKind = ...,
 30  ) -> NDArray[complex128]: ...
 31  
 32  def irfft(
 33      a: ArrayLike,
 34      n: None | int = ...,
 35      axis: int = ...,
 36      norm: _NormKind = ...,
 37  ) -> NDArray[float64]: ...
 38  
 39  # Input array must be compatible with `np.conjugate`
 40  def hfft(
 41      a: _ArrayLikeNumber_co,
 42      n: None | int = ...,
 43      axis: int = ...,
 44      norm: _NormKind = ...,
 45  ) -> NDArray[float64]: ...
 46  
 47  def ihfft(
 48      a: ArrayLike,
 49      n: None | int = ...,
 50      axis: int = ...,
 51      norm: _NormKind = ...,
 52  ) -> NDArray[complex128]: ...
 53  
 54  def fftn(
 55      a: ArrayLike,
 56      s: None | Sequence[int] = ...,
 57      axes: None | Sequence[int] = ...,
 58      norm: _NormKind = ...,
 59  ) -> NDArray[complex128]: ...
 60  
 61  def ifftn(
 62      a: ArrayLike,
 63      s: None | Sequence[int] = ...,
 64      axes: None | Sequence[int] = ...,
 65      norm: _NormKind = ...,
 66  ) -> NDArray[complex128]: ...
 67  
 68  def rfftn(
 69      a: ArrayLike,
 70      s: None | Sequence[int] = ...,
 71      axes: None | Sequence[int] = ...,
 72      norm: _NormKind = ...,
 73  ) -> NDArray[complex128]: ...
 74  
 75  def irfftn(
 76      a: ArrayLike,
 77      s: None | Sequence[int] = ...,
 78      axes: None | Sequence[int] = ...,
 79      norm: _NormKind = ...,
 80  ) -> NDArray[float64]: ...
 81  
 82  def fft2(
 83      a: ArrayLike,
 84      s: None | Sequence[int] = ...,
 85      axes: None | Sequence[int] = ...,
 86      norm: _NormKind = ...,
 87  ) -> NDArray[complex128]: ...
 88  
 89  def ifft2(
 90      a: ArrayLike,
 91      s: None | Sequence[int] = ...,
 92      axes: None | Sequence[int] = ...,
 93      norm: _NormKind = ...,
 94  ) -> NDArray[complex128]: ...
 95  
 96  def rfft2(
 97      a: ArrayLike,
 98      s: None | Sequence[int] = ...,
 99      axes: None | Sequence[int] = ...,
100      norm: _NormKind = ...,
101  ) -> NDArray[complex128]: ...
102  
103  def irfft2(
104      a: ArrayLike,
105      s: None | Sequence[int] = ...,
106      axes: None | Sequence[int] = ...,
107      norm: _NormKind = ...,
108  ) -> NDArray[float64]: ...