Gecko [ fernandoquevedodop.com ]

Name	Size	Permission
__pycache__	[ DIR ]	drwxr-xr-x
include	[ DIR ]	drwxr-xr-x
lib	[ DIR ]	drwxr-xr-x
tests	[ DIR ]	drwxr-xr-x
__init__.py	5.24 KB	-rw-r--r--
__init__.pyi	126 B	-rw-r--r--
_add_newdocs.py	187.41 KB	-rw-r--r--
_add_newdocs_scalars.py	8.59 KB	-rw-r--r--
_asarray.py	4.08 KB	-rw-r--r--
_asarray.pyi	1.89 KB	-rw-r--r--
_dtype.py	9.61 KB	-rw-r--r--
_dtype_ctypes.py	3.59 KB	-rw-r--r--
_exceptions.py	5.99 KB	-rw-r--r--
_internal.py	26.73 KB	-rw-r--r--
_internal.pyi	1.34 KB	-rw-r--r--
_methods.py	10.54 KB	-rw-r--r--
_multiarray_tests.cpython-310-...	122.6 KB	-rw-r--r--
_multiarray_umath.cpython-310-...	3.63 MB	-rw-r--r--
_operand_flag_tests.cpython-31...	14.2 KB	-rw-r--r--
_rational_tests.cpython-310-x8...	43.46 KB	-rw-r--r--
_simd.cpython-310-x86_64-linux...	1.91 MB	-rw-r--r--
_string_helpers.py	2.79 KB	-rw-r--r--
_struct_ufunc_tests.cpython-31...	14.38 KB	-rw-r--r--
_type_aliases.py	7.1 KB	-rw-r--r--
_type_aliases.pyi	520 B	-rw-r--r--
_ufunc_config.py	13.07 KB	-rw-r--r--
_ufunc_config.pyi	1.22 KB	-rw-r--r--
_umath_tests.cpython-310-x86_6...	34.8 KB	-rw-r--r--
arrayprint.py	60.18 KB	-rw-r--r--
arrayprint.pyi	4.56 KB	-rw-r--r--
cversions.py	347 B	-rw-r--r--
defchararray.py	68.1 KB	-rw-r--r--
einsumfunc.py	50.24 KB	-rw-r--r--
einsumfunc.pyi	3.62 KB	-rw-r--r--
fromnumeric.py	119.9 KB	-rw-r--r--
fromnumeric.pyi	7.83 KB	-rw-r--r--
function_base.py	18.57 KB	-rw-r--r--
function_base.pyi	1.44 KB	-rw-r--r--
generate_numpy_api.py	6.94 KB	-rw-r--r--
getlimits.py	19.31 KB	-rw-r--r--
machar.py	10.56 KB	-rw-r--r--
memmap.py	11.41 KB	-rw-r--r--
multiarray.py	54.01 KB	-rw-r--r--
numeric.py	74.93 KB	-rw-r--r--
numeric.pyi	4.76 KB	-rw-r--r--
numerictypes.py	16.91 KB	-rw-r--r--
numerictypes.pyi	2.85 KB	-rw-r--r--
overrides.py	7.94 KB	-rw-r--r--
records.py	36.58 KB	-rw-r--r--
setup.py	44.62 KB	-rw-r--r--
setup_common.py	19.31 KB	-rw-r--r--
shape_base.py	28.32 KB	-rw-r--r--
shape_base.pyi	1.04 KB	-rw-r--r--
umath.py	1.99 KB	-rw-r--r--
umath_tests.py	389 B	-rw-r--r--

Code Editor : einsumfunc.pyi

import sys
from typing import List, TypeVar, Optional, Any, overload, Union, Tuple, Sequence

from numpy import (
    ndarray,
    dtype,
    bool_,
    unsignedinteger,
    signedinteger,
    floating,
    complexfloating,
    number,
    _OrderKACF,
)
from numpy.typing import (
    _ArrayLikeBool_co,
    _ArrayLikeUInt_co,
    _ArrayLikeInt_co,
    _ArrayLikeFloat_co,
    _ArrayLikeComplex_co,
    _DTypeLikeBool,
    _DTypeLikeUInt,
    _DTypeLikeInt,
    _DTypeLikeFloat,
    _DTypeLikeComplex,
    _DTypeLikeComplex_co,
)

if sys.version_info >= (3, 8):
    from typing import Literal
else:
    from typing_extensions import Literal

_ArrayType = TypeVar(
    "_ArrayType",
    bound=ndarray[Any, dtype[Union[bool_, number[Any]]]],
)

_OptimizeKind = Union[
    None, bool, Literal["greedy", "optimal"], Sequence[Any]
]
_CastingSafe = Literal["no", "equiv", "safe", "same_kind"]
_CastingUnsafe = Literal["unsafe"]

__all__: List[str]

# TODO: Properly handle the `casting`-based combinatorics
# TODO: We need to evaluate the content `__subscripts` in order
# to identify whether or an array or scalar is returned. At a cursory
# glance this seems like something that can quite easilly be done with
# a mypy plugin.
# Something like `is_scalar = bool(__subscripts.partition("->")[-1])`
@overload
def einsum(
    __subscripts: str,
    *operands: _ArrayLikeBool_co,
    out: None = ...,
    dtype: Optional[_DTypeLikeBool] = ...,
    order: _OrderKACF = ...,
    casting: _CastingSafe = ...,
    optimize: _OptimizeKind = ...,
) -> Any: ...
@overload
def einsum(
    __subscripts: str,
    *operands: _ArrayLikeUInt_co,
    out: None = ...,
    dtype: Optional[_DTypeLikeUInt] = ...,
    order: _OrderKACF = ...,
    casting: _CastingSafe = ...,
    optimize: _OptimizeKind = ...,
) -> Any: ...
@overload
def einsum(
    __subscripts: str,
    *operands: _ArrayLikeInt_co,
    out: None = ...,
    dtype: Optional[_DTypeLikeInt] = ...,
    order: _OrderKACF = ...,
    casting: _CastingSafe = ...,
    optimize: _OptimizeKind = ...,
) -> Any: ...
@overload
def einsum(
    __subscripts: str,
    *operands: _ArrayLikeFloat_co,
    out: None = ...,
    dtype: Optional[_DTypeLikeFloat] = ...,
    order: _OrderKACF = ...,
    casting: _CastingSafe = ...,
    optimize: _OptimizeKind = ...,
) -> Any: ...
@overload
def einsum(
    __subscripts: str,
    *operands: _ArrayLikeComplex_co,
    out: None = ...,
    dtype: Optional[_DTypeLikeComplex] = ...,
    order: _OrderKACF = ...,
    casting: _CastingSafe = ...,
    optimize: _OptimizeKind = ...,
) -> Any: ...
@overload
def einsum(
    __subscripts: str,
    *operands: Any,
    casting: _CastingUnsafe,
    dtype: Optional[_DTypeLikeComplex_co] = ...,
    out: None = ...,
    order: _OrderKACF = ...,
    optimize: _OptimizeKind = ...,
) -> Any: ...
@overload
def einsum(
    __subscripts: str,
    *operands: _ArrayLikeComplex_co,
    out: _ArrayType,
    dtype: Optional[_DTypeLikeComplex_co] = ...,
    order: _OrderKACF = ...,
    casting: _CastingSafe = ...,
    optimize: _OptimizeKind = ...,
) -> _ArrayType: ...
@overload
def einsum(
    __subscripts: str,
    *operands: Any,
    out: _ArrayType,
    casting: _CastingUnsafe,
    dtype: Optional[_DTypeLikeComplex_co] = ...,
    order: _OrderKACF = ...,
    optimize: _OptimizeKind = ...,
) -> _ArrayType: ...

# NOTE: `einsum_call` is a hidden kwarg unavailable for public use.
# It is therefore excluded from the signatures below.
# NOTE: In practice the list consists of a `str` (first element)
# and a variable number of integer tuples.
def einsum_path(
    __subscripts: str,
    *operands: _ArrayLikeComplex_co,
    optimize: _OptimizeKind = ...,
) -> Tuple[List[Any], str]: ...

${this.title}

Code Editor : einsumfunc.pyi