You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
176 lines
4.6 KiB
Python
176 lines
4.6 KiB
Python
from collections.abc import Sequence
|
|
from typing import Any, SupportsIndex, TypeVar, overload
|
|
|
|
from numpy import _CastingKind, generic
|
|
from numpy._typing import ArrayLike, DTypeLike, NDArray, _ArrayLike, _DTypeLike
|
|
|
|
__all__ = [
|
|
"atleast_1d",
|
|
"atleast_2d",
|
|
"atleast_3d",
|
|
"block",
|
|
"hstack",
|
|
"stack",
|
|
"unstack",
|
|
"vstack",
|
|
]
|
|
|
|
_ScalarT = TypeVar("_ScalarT", bound=generic)
|
|
_ScalarT1 = TypeVar("_ScalarT1", bound=generic)
|
|
_ScalarT2 = TypeVar("_ScalarT2", bound=generic)
|
|
_ArrayT = TypeVar("_ArrayT", bound=NDArray[Any])
|
|
|
|
###
|
|
|
|
@overload
|
|
def atleast_1d(a0: _ArrayLike[_ScalarT], /) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def atleast_1d(a0: _ArrayLike[_ScalarT1], a1: _ArrayLike[_ScalarT2], /) -> tuple[NDArray[_ScalarT1], NDArray[_ScalarT2]]: ...
|
|
@overload
|
|
def atleast_1d(a0: _ArrayLike[_ScalarT], a1: _ArrayLike[_ScalarT], /, *arys: _ArrayLike[_ScalarT]) -> tuple[NDArray[_ScalarT], ...]: ...
|
|
@overload
|
|
def atleast_1d(a0: ArrayLike, /) -> NDArray[Any]: ...
|
|
@overload
|
|
def atleast_1d(a0: ArrayLike, a1: ArrayLike, /) -> tuple[NDArray[Any], NDArray[Any]]: ...
|
|
@overload
|
|
def atleast_1d(a0: ArrayLike, a1: ArrayLike, /, *ai: ArrayLike) -> tuple[NDArray[Any], ...]: ...
|
|
|
|
#
|
|
@overload
|
|
def atleast_2d(a0: _ArrayLike[_ScalarT], /) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def atleast_2d(a0: _ArrayLike[_ScalarT1], a1: _ArrayLike[_ScalarT2], /) -> tuple[NDArray[_ScalarT1], NDArray[_ScalarT2]]: ...
|
|
@overload
|
|
def atleast_2d(a0: _ArrayLike[_ScalarT], a1: _ArrayLike[_ScalarT], /, *arys: _ArrayLike[_ScalarT]) -> tuple[NDArray[_ScalarT], ...]: ...
|
|
@overload
|
|
def atleast_2d(a0: ArrayLike, /) -> NDArray[Any]: ...
|
|
@overload
|
|
def atleast_2d(a0: ArrayLike, a1: ArrayLike, /) -> tuple[NDArray[Any], NDArray[Any]]: ...
|
|
@overload
|
|
def atleast_2d(a0: ArrayLike, a1: ArrayLike, /, *ai: ArrayLike) -> tuple[NDArray[Any], ...]: ...
|
|
|
|
#
|
|
@overload
|
|
def atleast_3d(a0: _ArrayLike[_ScalarT], /) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def atleast_3d(a0: _ArrayLike[_ScalarT1], a1: _ArrayLike[_ScalarT2], /) -> tuple[NDArray[_ScalarT1], NDArray[_ScalarT2]]: ...
|
|
@overload
|
|
def atleast_3d(a0: _ArrayLike[_ScalarT], a1: _ArrayLike[_ScalarT], /, *arys: _ArrayLike[_ScalarT]) -> tuple[NDArray[_ScalarT], ...]: ...
|
|
@overload
|
|
def atleast_3d(a0: ArrayLike, /) -> NDArray[Any]: ...
|
|
@overload
|
|
def atleast_3d(a0: ArrayLike, a1: ArrayLike, /) -> tuple[NDArray[Any], NDArray[Any]]: ...
|
|
@overload
|
|
def atleast_3d(a0: ArrayLike, a1: ArrayLike, /, *ai: ArrayLike) -> tuple[NDArray[Any], ...]: ...
|
|
|
|
#
|
|
@overload
|
|
def vstack(
|
|
tup: Sequence[_ArrayLike[_ScalarT]],
|
|
*,
|
|
dtype: None = ...,
|
|
casting: _CastingKind = ...
|
|
) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def vstack(
|
|
tup: Sequence[ArrayLike],
|
|
*,
|
|
dtype: _DTypeLike[_ScalarT],
|
|
casting: _CastingKind = ...
|
|
) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def vstack(
|
|
tup: Sequence[ArrayLike],
|
|
*,
|
|
dtype: DTypeLike = ...,
|
|
casting: _CastingKind = ...
|
|
) -> NDArray[Any]: ...
|
|
|
|
@overload
|
|
def hstack(
|
|
tup: Sequence[_ArrayLike[_ScalarT]],
|
|
*,
|
|
dtype: None = ...,
|
|
casting: _CastingKind = ...
|
|
) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def hstack(
|
|
tup: Sequence[ArrayLike],
|
|
*,
|
|
dtype: _DTypeLike[_ScalarT],
|
|
casting: _CastingKind = ...
|
|
) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def hstack(
|
|
tup: Sequence[ArrayLike],
|
|
*,
|
|
dtype: DTypeLike = ...,
|
|
casting: _CastingKind = ...
|
|
) -> NDArray[Any]: ...
|
|
|
|
@overload
|
|
def stack(
|
|
arrays: Sequence[_ArrayLike[_ScalarT]],
|
|
axis: SupportsIndex = ...,
|
|
out: None = ...,
|
|
*,
|
|
dtype: None = ...,
|
|
casting: _CastingKind = ...
|
|
) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def stack(
|
|
arrays: Sequence[ArrayLike],
|
|
axis: SupportsIndex = ...,
|
|
out: None = ...,
|
|
*,
|
|
dtype: _DTypeLike[_ScalarT],
|
|
casting: _CastingKind = ...
|
|
) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def stack(
|
|
arrays: Sequence[ArrayLike],
|
|
axis: SupportsIndex = ...,
|
|
out: None = ...,
|
|
*,
|
|
dtype: DTypeLike = ...,
|
|
casting: _CastingKind = ...
|
|
) -> NDArray[Any]: ...
|
|
@overload
|
|
def stack(
|
|
arrays: Sequence[ArrayLike],
|
|
axis: SupportsIndex,
|
|
out: _ArrayT,
|
|
*,
|
|
dtype: DTypeLike | None = None,
|
|
casting: _CastingKind = "same_kind",
|
|
) -> _ArrayT: ...
|
|
@overload
|
|
def stack(
|
|
arrays: Sequence[ArrayLike],
|
|
axis: SupportsIndex = 0,
|
|
*,
|
|
out: _ArrayT,
|
|
dtype: DTypeLike | None = None,
|
|
casting: _CastingKind = "same_kind",
|
|
) -> _ArrayT: ...
|
|
|
|
@overload
|
|
def unstack(
|
|
array: _ArrayLike[_ScalarT],
|
|
/,
|
|
*,
|
|
axis: int = ...,
|
|
) -> tuple[NDArray[_ScalarT], ...]: ...
|
|
@overload
|
|
def unstack(
|
|
array: ArrayLike,
|
|
/,
|
|
*,
|
|
axis: int = ...,
|
|
) -> tuple[NDArray[Any], ...]: ...
|
|
|
|
@overload
|
|
def block(arrays: _ArrayLike[_ScalarT]) -> NDArray[_ScalarT]: ...
|
|
@overload
|
|
def block(arrays: ArrayLike) -> NDArray[Any]: ...
|