Более быстрое скалярное произведение матрицы CSR и ndarray - PullRequest
0 голосов
/ 08 февраля 2019

Я пытаюсь получить функцию произведения быстрой точки для умножения разреженной матрицы (3 * 3) и массива nd (1 * 3) таким образом, чтобы каждая строка матрицы получала произведение точки с массивом nd, чтобы получить(3 * 1) массив.

Моя текущая реализация состоит в том, чтобы получить каждую строку матрицы и затем выполнить скалярное произведение, но для увеличения размера матрицы она становится слишком медленной.

row = np.array([0, 0, 1, 2, 2, 2])
col = np.array([0, 2, 2, 0, 1, 2])
data = np.array([1, 2, 3, 4, 5, 6])
matrix =csr_matrix((data, (row, col)), shape=(3, 3))
secondArray=np.random.rand((1, 3))

for idx, x in enumerate(matrix):
    X_arr=X.getrow(idx).toarray()
    prod=np.dot(np.array(X_arr[0]), secondArray)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...