Function reduce_sum

imperative/python/megengine/functional/einsum.py:441–447

(x, axis)

                return shape

        def reduce_sum(x, axis):
            if len(axis) == 0:
                return x
            for i in reversed(axis):
                x = f(builtin.Reduce(mode="sum", axis=i), x)
            x = f(builtin.RemoveAxis(axis=[*reversed(axis)]), x)
            return x

        ctx.dims2val = dims2val
        ctx.reduce = reduce_sum