Векторизация умножения на разные константы - PullRequest
0 голосов
/ 25 февраля 2020

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

    for i in range(10):
        expansions[i,:,:] += c[i,k]*(matrix)

Теперь, это простой способ сделать это, но мне было интересно, есть ли какой-то способ сделать это так:

        expansions[:,:,:] += c[:,k]*(matrix)

Где я применяю каждую константу индивидуально к «матрице». Я пытался сложить много «матрицы» поверх себя, чтобы создать что-то размером (10xNxN), но это не сработало. Есть идеи?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...