Function partition_num

ge/utils.py:44–48

(num, workers)

def partition_num(num, workers):
    if num % workers == 0:
        return [num // workers] * workers
    else:
        return [num // workers] * workers + [num % workers]