Индексация np.einsum для суммирования - PullRequest
0 голосов
/ 10 января 2019

Я пытаюсь выполнить следующее суммирование вектора (v) и двух матриц (B), используя np.einsum

A [i, j] = (сумма от k = 1 до N) v [k] * B [k, i] * B [k, j]

A = np.einsum('k,ki,kj->ij',v,B,B)

Я ожидаю, что это суммирование по индексу k приведет к матрице i x j, но я не уверен, выполняет ли это суммирование или просто умножение

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