ускорить индексирование подмассивов в Matlab?

Question

Как ускорить выполнение следующих операций? Узкое место является третьей линией, хотя для больших размеров A1, четвертая линия довольно быстрая. Делает ли третья строка копию A (b, b), которая хранится в A1?

A = randn(1000,1000); 
b = [67 145 200 185 11 166 80 137 163 132 133 19]; %random 
A1 = A(b,b); 
v=A1*A(2,b)'; 

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

v=A(b,b);*A(2,b)'; 

Answer 1

Смотрите, если это делает его быстрее в качестве замены для третьей линии, что вы утверждаете быть узким местом -

[x,y]= ndgrid(b,b); 
A1 = A((y-1)*size(A,1)+x); 

Или

A1 = A(bsxfun(@plus,(b-1)*size(A,1),b')); 

---

Edit: После профилирования указанными выше кодами производительность во время выполнения не выглядит доказано много. Вы упомянули в комментариях, что вы используете эти коды в цикле несколько раз и b различается. Если количество циклов было небольшим, а b было константой между этими итерациями цикла, можно было бы подумать о выполнении всех этих умножений матричных векторов по всем итерациям в одно большое умножение матричной матрицы, но здесь это не так. Поэтому я бы сказал, что на этом этапе узкое место с indexing, возможно, останется.