Вычисление следа большой разреженной матрицы - PullRequest
0 голосов
/ 26 января 2019

У меня есть разреженная большая матрица (линейный размер 2 * 10 ^ 6), для которой я хочу вычислить ее след. Вычисление этого грубой силы занимает 16 секунд, чтобы получить доступ к каждому диагональному элементу (следовательно, я мог бы сделать это за ГОД!).

Я думал о том, чтобы сохранить его на диск с помощью scipy.io.mmwrite и прочитать его с кодом C ++, который должен быть намного быстрее. Однако я не могу найти какой-либо пакет, который мог бы помочь мне сделать это.

Любые предложения будут высоко оценены!

Спасибо.

1 Ответ

0 голосов
/ 28 января 2019

Решение было простым: np.array (Mat.diagonal ()). Sum ()

Спасибо, @hpaulj!

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