Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.11/site-packages/opt_einsum/typing.py: 0%
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1"""Types used in the opt_einsum package."""
3from collections import namedtuple
4from typing import Any, Callable, Collection, Dict, FrozenSet, List, Literal, Optional, Tuple, Union
6TensorShapeType = Tuple[int, ...]
7PathType = Collection[TensorShapeType]
9ArrayType = Any
11ArrayIndexType = FrozenSet[str]
12ArrayShaped = namedtuple("ArrayShaped", ["shape"])
14ContractionListType = List[Tuple[Any, ArrayIndexType, str, Optional[Tuple[str, ...]], Union[str, bool]]]
15PathSearchFunctionType = Callable[[List[ArrayIndexType], ArrayIndexType, Dict[str, int], Optional[int]], PathType]
17# Contract kwargs
18OptimizeKind = Union[
19 None,
20 bool,
21 Literal[
22 "optimal", "dp", "greedy", "random-greedy", "random-greedy-128", "branch-all", "branch-2", "auto", "auto-hq"
23 ],
24 PathType,
25 PathSearchFunctionType,
26]
27BackendType = Literal["auto", "object", "autograd", "cupy", "dask", "jax", "theano", "tensorflow", "torch", "libjax"]