| | from typing import Any, TypeVar, overload, Literal as L |
| |
|
| | from numpy import generic, integer, floating, complexfloating |
| | from numpy._typing import ( |
| | NDArray, |
| | ArrayLike, |
| | _ShapeLike, |
| | _ArrayLike, |
| | _ArrayLikeFloat_co, |
| | _ArrayLikeComplex_co, |
| | ) |
| |
|
| | _SCT = TypeVar("_SCT", bound=generic) |
| |
|
| | __all__: list[str] |
| |
|
| | @overload |
| | def fftshift(x: _ArrayLike[_SCT], axes: None | _ShapeLike = ...) -> NDArray[_SCT]: ... |
| | @overload |
| | def fftshift(x: ArrayLike, axes: None | _ShapeLike = ...) -> NDArray[Any]: ... |
| |
|
| | @overload |
| | def ifftshift(x: _ArrayLike[_SCT], axes: None | _ShapeLike = ...) -> NDArray[_SCT]: ... |
| | @overload |
| | def ifftshift(x: ArrayLike, axes: None | _ShapeLike = ...) -> NDArray[Any]: ... |
| |
|
| | @overload |
| | def fftfreq( |
| | n: int | integer[Any], |
| | d: _ArrayLikeFloat_co = ..., |
| | device: None | L["cpu"] = ..., |
| | ) -> NDArray[floating[Any]]: ... |
| | @overload |
| | def fftfreq( |
| | n: int | integer[Any], |
| | d: _ArrayLikeComplex_co = ..., |
| | device: None | L["cpu"] = ..., |
| | ) -> NDArray[complexfloating[Any, Any]]: ... |
| |
|
| | @overload |
| | def rfftfreq( |
| | n: int | integer[Any], |
| | d: _ArrayLikeFloat_co = ..., |
| | device: None | L["cpu"] = ..., |
| | ) -> NDArray[floating[Any]]: ... |
| | @overload |
| | def rfftfreq( |
| | n: int | integer[Any], |
| | d: _ArrayLikeComplex_co = ..., |
| | device: None | L["cpu"] = ..., |
| | ) -> NDArray[complexfloating[Any, Any]]: ... |
| |
|
