Skip to main content

Module parser

Module parser 

Source

Functions§

convert_subscripts
Convert user custom subscripts list to subscript string according to symbol_map.
find_output_shape
Find the output shape for given inputs, shapes and output string, taking into account broadcasting.
find_output_str
Find the output string for the inputs subscripts under canonical Einstein summation rules. That is, repeated indices are summed over by default.
gen_unused_symbols
Generate n unused symbols not in used.
get_symbol
Get the symbol corresponding to int i - runs through the usual 52 letters before resorting to unicode characters, starting at chr(192) and skipping surrogates.
parse_einsum_input
A reproduction of einsum’s parsing logic in Rust (originally from Python/Numpy).