You are viewing an older version (2.4.3). Go to latest (2.4.4)
{ } Raw JSON

bundles / numpy 2.4.3 / numpy / _core / einsumfunc / _parse_einsum_input

function

numpy._core.einsumfunc:_parse_einsum_input

source: /numpy/_core/einsumfunc.py :445

Signature

def   _parse_einsum_input ( operands )

Summary

A reproduction of einsum c side einsum parsing in python.

Returns

input_strings : str

Parsed input strings

output_string : str

Parsed output string

operands : list of array_like

The operands to use in the numpy contraction

Examples

The operand list is simplified to reduce printing:
np.random.seed(123)
a = np.random.rand(4, 4)
b = np.random.rand(4, 4, 4)
_parse_einsum_input(('...a,...a->...', a, b))
_parse_einsum_input((a, [Ellipsis, 0], b, [Ellipsis, 0]))

Aliases

  • numpy._core.einsumfunc._parse_einsum_input