Eigen - диагональное обновление разреженной матрицы

Question

Какой самый быстрый способ редактировать диагональ разреженной матрицы в Eigen? Я уже заполнил его с помощью триплетов, но иногда мне нужно изменить все диагональные значения (они уже были установлены на ненулевое значение).

Answer 1

Если приемлемо, вы могли бы просто использовать их индексы, как

sp_mat(i, i) 

Поскольку каждый доступ включает в себя бинарный поиск, производительность не может быть удовлетворено. В этом случае вы можете добиться большей производительности с некоторыми ограничениями.

1. Вы не меняете основной макет матрицы, например. добавление/удаление nnzs, сжатие;
2. Вы будете обращаться к диагонали повторно.

С этими ограничениями вы можете хранить указатели диагональных компонентов, чтобы вы могли получить доступ к данным позже в O (1) раз. Вы можете получить указатель с

&sp_mat(i, i) 

Answer 2

Для завершения kangshiyin ответа, вы можете также редактировать диагонали через плотные векторные выражения, например, чтобы добавить 1 к каждому диагональному элементу:

sp_mat.diagonal().array() += 1; 

Предупреждения: это будет работать только в том случае, если диагональные элементы уже существуют, что имеет место в исходном вопросе.