Распараллеливание матричных вычислений с помощью Numba - PullRequest
Новая
       94

Распараллеливание матричных вычислений с помощью Numba

0 голосов
/ 07 февраля 2020

Я пытаюсь распараллелить следующую функцию в Python 3.6: 

def get_Q(predictors, d):
    Q = 0.
    for i in range(len(predictors)):
        for j in range(i+1, len(predictors)):
            ci = predictors[i,:]
            cj = predictors[j,:]
            Q += product_cols(ci, cj, d[i], d[j])
    return Q

, где:
predictors[,] - массив ввода, с n - строки и p -функции, а
d[] - это массив n.

Функция product_cols() определяется как : 

def product_cols(ci, cj, di, dj):
    c = ci-cj
    return np.matmul(np.transpose(np.matrix(c)), np.matrix(c))*((di-dj)**2)

где:
ci[], cj[] - это p -размеры и
di, dj - действительные числа.

Может ли кто-нибудь помочь мне просто распараллелить это?

...