| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235 |
- from typing import (
- Literal as L,
- overload,
- Any,
- SupportsIndex,
- TypeVar,
- )
- from numpy import floating, complexfloating, generic
- from numpy._typing import (
- NDArray,
- DTypeLike,
- _DTypeLike,
- _ArrayLikeFloat_co,
- _ArrayLikeComplex_co,
- )
- __all__ = ["logspace", "linspace", "geomspace"]
- _SCT = TypeVar("_SCT", bound=generic)
- @overload
- def linspace(
- start: _ArrayLikeFloat_co,
- stop: _ArrayLikeFloat_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- retstep: L[False] = ...,
- dtype: None = ...,
- axis: SupportsIndex = ...,
- *,
- device: L["cpu"] | None = ...,
- ) -> NDArray[floating]: ...
- @overload
- def linspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- retstep: L[False] = ...,
- dtype: None = ...,
- axis: SupportsIndex = ...,
- *,
- device: L["cpu"] | None = ...,
- ) -> NDArray[complexfloating]: ...
- @overload
- def linspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex,
- endpoint: bool,
- retstep: L[False],
- dtype: _DTypeLike[_SCT],
- axis: SupportsIndex = ...,
- *,
- device: L["cpu"] | None = ...,
- ) -> NDArray[_SCT]: ...
- @overload
- def linspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- retstep: L[False] = ...,
- *,
- dtype: _DTypeLike[_SCT],
- axis: SupportsIndex = ...,
- device: L["cpu"] | None = ...,
- ) -> NDArray[_SCT]: ...
- @overload
- def linspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- retstep: L[False] = ...,
- dtype: DTypeLike = ...,
- axis: SupportsIndex = ...,
- *,
- device: L["cpu"] | None = ...,
- ) -> NDArray[Any]: ...
- @overload
- def linspace(
- start: _ArrayLikeFloat_co,
- stop: _ArrayLikeFloat_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- *,
- retstep: L[True],
- dtype: None = ...,
- axis: SupportsIndex = ...,
- device: L["cpu"] | None = ...,
- ) -> tuple[NDArray[floating], floating]: ...
- @overload
- def linspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- *,
- retstep: L[True],
- dtype: None = ...,
- axis: SupportsIndex = ...,
- device: L["cpu"] | None = ...,
- ) -> tuple[NDArray[complexfloating], complexfloating]: ...
- @overload
- def linspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- *,
- retstep: L[True],
- dtype: _DTypeLike[_SCT],
- axis: SupportsIndex = ...,
- device: L["cpu"] | None = ...,
- ) -> tuple[NDArray[_SCT], _SCT]: ...
- @overload
- def linspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- *,
- retstep: L[True],
- dtype: DTypeLike = ...,
- axis: SupportsIndex = ...,
- device: L["cpu"] | None = ...,
- ) -> tuple[NDArray[Any], Any]: ...
- @overload
- def logspace(
- start: _ArrayLikeFloat_co,
- stop: _ArrayLikeFloat_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- base: _ArrayLikeFloat_co = ...,
- dtype: None = ...,
- axis: SupportsIndex = ...,
- ) -> NDArray[floating]: ...
- @overload
- def logspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- base: _ArrayLikeComplex_co = ...,
- dtype: None = ...,
- axis: SupportsIndex = ...,
- ) -> NDArray[complexfloating]: ...
- @overload
- def logspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex,
- endpoint: bool,
- base: _ArrayLikeComplex_co,
- dtype: _DTypeLike[_SCT],
- axis: SupportsIndex = ...,
- ) -> NDArray[_SCT]: ...
- @overload
- def logspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- base: _ArrayLikeComplex_co = ...,
- *,
- dtype: _DTypeLike[_SCT],
- axis: SupportsIndex = ...,
- ) -> NDArray[_SCT]: ...
- @overload
- def logspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- base: _ArrayLikeComplex_co = ...,
- dtype: DTypeLike = ...,
- axis: SupportsIndex = ...,
- ) -> NDArray[Any]: ...
- @overload
- def geomspace(
- start: _ArrayLikeFloat_co,
- stop: _ArrayLikeFloat_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- dtype: None = ...,
- axis: SupportsIndex = ...,
- ) -> NDArray[floating]: ...
- @overload
- def geomspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- dtype: None = ...,
- axis: SupportsIndex = ...,
- ) -> NDArray[complexfloating]: ...
- @overload
- def geomspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex,
- endpoint: bool,
- dtype: _DTypeLike[_SCT],
- axis: SupportsIndex = ...,
- ) -> NDArray[_SCT]: ...
- @overload
- def geomspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- *,
- dtype: _DTypeLike[_SCT],
- axis: SupportsIndex = ...,
- ) -> NDArray[_SCT]: ...
- @overload
- def geomspace(
- start: _ArrayLikeComplex_co,
- stop: _ArrayLikeComplex_co,
- num: SupportsIndex = ...,
- endpoint: bool = ...,
- dtype: DTypeLike = ...,
- axis: SupportsIndex = ...,
- ) -> NDArray[Any]: ...
- def add_newdoc(
- place: str,
- obj: str,
- doc: str | tuple[str, str] | list[tuple[str, str]],
- warn_on_python: bool = ...,
- ) -> None: ...
|