|
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]]: ... |
|
|
