Speed Up Array Allocation

Question

Moe Szyslak 2020-1-21

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/501350-speed-up-array-allocation

编辑： Andrey Porokhnyuk 2020-1-22

Hi, All -- I have a problem containing a double sum that I am currently solving with nested for loops (below). It seems to me that there should be a good way to vectorize/otherwise speed up these calculations. On a modern workstation (Windows 10) using my data set of size [N x m], 100 iterations of this calculation takes about 7 s. Not horrible, but I need to apply this to many, many data sets, some larger, some smaller. I just don't want to leave any efficiencies on the table. Thanks, in advance.

Matt

tic
% Some preliminaries
AA = zeros(N,3);
for i = 1:N             % Should be able to vectorize...?
    for j = 1:m
        AA(i,1) = AA(i,1) + exp((-((2*j-1)*pi/2)^2)*cV1(k)*i);
        AA(i,2) = AA(i,2) + ((2*j-1)^2)*exp((-((2*j-1)*pi/2)^2)*cV1(k)*i);
        AA(i,3) = AA(i,3) + ((2*j-1)^-2)*exp((-((2*j-1)*pi/2)^2)*cV1(k)*i);
    end
end
vec1 = AA(:,1);
vec2 = AA(:,2);
vec3 = AA(:,3);
toc

2 个评论
显示无隐藏无

James Tursa 2020-1-21

What version of MATLAB are you using? How large are m and N?

Moe Szyslak 2020-1-21

I am in R2019b. [M x n] is approximately [6000 x 100], but could be larger. It takes about 0.068 s to build the array once and I go through the calculations 100 times, iterating over k.

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Walter Roberson 2020-1-21

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/501350-speed-up-array-allocation#answer_411305

在 MATLAB Online 中打开

j = 1 : m;
i = 1 : N;
temp = exp((-((2*j-1)*pi/2).^2)*cV1(k).*i.');
vec1 = sum(temp,2);
vec2 = sum((2*j-1).^2 .* temp, 2);
vec3 = sum((2*j-1).^-2 .* temp, 2);

1 个评论
显示 -1更早的评论隐藏 -1更早的评论

Moe Szyslak 2020-1-21

Brilliant! This gets me a 12-fold speed increase from the previous version. I tried to program essentially what you did, but I was going wrong somewhere because my array sizes were never correct.

Appreciate the help -- thank you very much.

请先登录，再进行评论。

Answer 2

Andrey Porokhnyuk 2020-1-21

1
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/501350-speed-up-array-allocation#answer_411319

编辑：Andrey Porokhnyuk 2020-1-22

in the past (arount the year 2008) I was speaking with a coworker, who found that functions written in separate files are executed much faster. He was solving nonlinear LLG problems, and it seemed like giving a good boost. can not confirm it because I was using octave and scilab from that time, things are different there.

In Octave I would try declaring types explicitly before allocating elements. just think about it rationally - automatic vectors used for general mat type have pages long templates, so the memory structure should be quite complex. when you have simple int's doubles and singles, interpretter knows exactly how many bytes are necessary for every element from the beginning. It may help cutting corners.