MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / EinsumContext

Class EinsumContext

imperative/python/megengine/functional/einsum.py:22–29  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

20
21
22class EinsumContext:
23 dims2val: Callable[[str], Any]
24 reshape: Callable[[Any, tuple], Any]
25 broadcast: Callable[[Any, tuple], Any]
26 transpose: Callable[[Any, List[int]], Any]
27 reduce: Callable[[Any, List[int]], Any]
28 matmul: Callable[[Any, Any], Any]
29 diag_plane: Callable[[Any, int, List[int]], Any]
30
31
32class EinsumOperand:

Callers 2

einsum_interpretFunction · 0.85
einsumFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected