Parallelizing four nested loops in Python

后端 未结 3 615
予麋鹿
予麋鹿 2020-12-28 09:40

I have a fairly straightforward nested for loop that iterates over four arrays:

for a in a_grid:
    for b in b_grid:
        for c in c_grid:
            fo         


        
3条回答
  •  心在旅途
    2020-12-28 10:38

    The number of jobs is not related to the number of nested loops. In that other answer, it happened to be n_jobs=2 and 2 loops, but the two are completely unrelated.

    Think of it this way: You have a bunch of function calls to make; in your case (unrolling the loops):

    do_some_stuff(0,0,0,0)
    do_some_stuff(0,0,0,1)
    do_some_stuff(0,0,0,2)
    do_some_stuff(0,0,1,0)
    do_some_stuff(0,0,1,1)
    do_some_stuff(0,0,1,2)
    ...
    

    and you want to distribute those function calls across some number of jobs. You could use 2 jobs, or 10, or 100, it doesn't matter. Parallel takes care of distributing the work for you.

提交回复
热议问题