У меня есть критически важный для производительности фрагмент кода, где мне нужно проверить один массив на значения ниже порога, а затем условно установить значения двух других массивов. Мой код выглядит так:
#include <Eigen/Dense>
int main(){
Eigen::ArrayXXd
a (1, 100),
b (2, 100),
c (3, 100);
a.setRandom();
b.setRandom();
c.setRandom();
constexpr double minVal { 1e-8 };
/* the code segment in question */
/* option 1 */
for ( int i=0; i<2; ++i ){
b.row(i) = (a < minVal).select( 0, c.row(i+1) / a );
c.row(i+1) = (a < minVal).select( 0, c.row(i+1) );
}
/* option 2, which is slower */
b = (a < minVal).replicate(2,1).select( 0, c.bottomRows(2) / a.replicate(2,1) );
c.bottomRows(2) = (a < minVal).replicate(2,1).select( 0, c.bottomRows(2) );
return 0;
}
Массив a, значения которого проверяются на достижение порога minVal, имеет одну строку и динамическое количество столбцов. Два других массива b и c имеют две и три строки соответственно и такое же количество столбцов, что и a.
Теперь я хотел бы реализовать вышеупомянутую логику более eigen способом, без этого цикла в варианте 1, потому что обычно у eigen есть хитрости в рукаве для повышения производительности, которые я никогда не могу надеяться сопоставить при написании сырых циклов. Однако единственным выходом, который я мог придумать, был вариант 2, который заметно медленнее, чем вариант 1.
Каким будет правильный и эффективный способ сделать это? Или петля уже мой лучший вариант?
