Как вернуть генератор с помощью joblib.parallel ()? ⇐ Python

[Anonymous]

У меня есть кусок кода ниже, где

Код: Выделить всё

joblib.Parallel()

[/b] возвращает список.

Код: Выделить всё

import numpy as np
from joblib import Parallel, delayed

lst = [[0.0, 1, 2], [3, 4, 5], [6, 7, 8]]
arr = np.array(lst)
w, v = np.linalg.eigh(arr)

def proj_func(i):
return np.dot(v[:,i].reshape(-1, 1), v[:,i].reshape(1, -1))

proj = Parallel(n_jobs=-1)(delayed(proj_func)(i) for i in range(len(w)))
< /code>

Вместо списка, как вернуть генератор с использованием joblib.parallel () < /code>?import numpy as np
from joblib import Parallel, delayed

lst = [[0.0, 1, 2], [3, 4, 5], [6, 7, 8]]
arr = np.array(lst)
w, v = np.linalg.eigh(arr)

def proj_func(i):
yield np.dot(v[:,i].reshape(-1, 1), v[:,i].reshape(1, -1))

proj = Parallel(n_jobs=-1)(delayed(proj_func)(i) for i in range(len(w)))
< /code>

Но я получаю эту ошибку: < /p>

TypeError: can't pickle generator objects
< /code>

Я что -то упускаю? Как мне это исправить? Мой основной прирост здесь - уменьшить память, так как Proj 

может стать очень большим, поэтому я бы просто хотел позвонить каждому генератору в списке по одному.

Подробнее здесь: https://stackoverflow.com/questions/605 ... b-parallel