Function safe_zip

imperative/python/megengine/xla/utils.py:32–37

(*args)

# zip with args length check
def safe_zip(*args):
    args = list(map(list, args))
    n = len(args[0])
    for arg in args[1:]:
        assert len(arg) == n, f"length mismatch: {list(map(len, args))}"
    return list(zip(*args))


# ((x, a), (y, b), (z, c)) -> (x, y, z), (a, b, c)