einsum

class Einsum(expr, shapes)

Bases: object

einsum(expr, *args, **kwargs)

ordered_indices(expr, shapes)

find a better order to reduce transpose.

remove_size1(expr, *args, extra=None)

remove order independent indices (size 1)

replace_ellipsis(expr, shapes)

replace_none_in_shape(x, num=-1)

symbol_generate(base_map)

tensor_einsum_reduce_sum(expr, *args, order)

“abe,bcf->acef” =reshape=> “ab1e1,1bc1f->acef” =product=> “abcef->acef” =reduce_sum=> “acef”