Route einsum

imperative/python/megengine/functional/einsum.py:None–None

subgraph_fn(
        "Einsum", dtype=dtype, device=device, nr_inputs=len(input_ndims), gopt_level=2
    )

import os
from functools import lru_cache, partial, reduce
from typing import Any, Callable, List, Optional, Tuple