Function dim2val

imperative/python/megengine/functional/einsum.py:379–386

(dim: str)

    dim2val_cache = {}

    def dim2val(dim: str):
        if dim in dim2val_cache:
            return dim2val_cache[dim]
        for i, shape in enumerate(shapes):
            if dim in shape:
                dim2val_cache[dim] = inputs[i].shape[shape.index(dim)]
                break
        return dim2val_cache[dim]

    def dims2val(dims: str):
        if len(dims) == 0: