Nested for loops newbie question

1 次查看（过去 30 天）
ttopal
ttopal 2016-8-1
编辑： Stephen23 2016-8-1
Hi,
I know this might be nested for loops question 1000. I have read most of other posts, even the newbie questions seems little bit advanced. My code is simple, yet I don't know where to look at to improve its speed. This loops takes more than 170 sec for N=512 M=128.
Thanks
for n = 1:M
for m = 1:M
for k = 1:N
for l = 1:N
temp=temp+fun(k,l)*exp(-i*2*pi*( n*t(k)/T + m*t(l)/T));
end
end
c(n,m) = coef*temp;
temp=0;
end
end
  2 个评论
Stephen23
Stephen23 2016-8-1
What are fun, t, T, coef, and temp ?
ttopal
ttopal 2016-8-1
Hi Stephen, I am trying to solve numerical 2D integration, I have parameters t(k) and t(l) as intervals and fun is 2DArray values as fun(t(k),t(l)). T is period. Coefficient is double. Temp is complex for storing temporary sum.

请先登录，再进行评论。

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

采纳的回答

Stephen23
Stephen23 2016-8-1
编辑：Stephen23 2016-8-1
So you have some nested loops and you want to make your code faster.
The first step is to read this page:
http://www.mathworks.com/help/matlab/matlab_prog/techniques-for-improving-performance.html
and in particular you need to understand when arrays need to be preallocated before loops:
http://www.mathworks.com/help/matlab/matlab_prog/preallocating-arrays.html
and the importance of code vectorization:
http://www.mathworks.com/help/matlab/matlab_prog/vectorization.html
In general there is no magic solution for speeding up some looped code: it depends on the operations being performed. In your case the operations are quite simple, and could be vectorized quite easily. Here is a vectorized version that runs in a quarter of the time (I did not test it on the full size matrices):
% Constants:
Mn = 128;
Nn = 51;
%Mn = 128;
%Nn = 512;
temp = 0;
fun = randi(9,Nn,Nn);
t = 1:Nn;
T = 10;
coef = 3;
old = NaN(Mn);
% Your code:
tic
for M1 = 1:Mn
for M2 = 1:Mn
for N1 = 1:Nn
for N2 = 1:Nn
temp=temp+fun(N1,N2)*exp(-1i*2*pi*( M1*t(N1)/T + M2*t(N2)/T));
end
end
old(M1,M2) = coef*temp;
temp=0;
end
end
toc
% Vectorized version:
tic
[N1m,~,M1m,~] = ndgrid(1:Nn,1,1:Mn,1);
[~,N2m,~,M2m] = ndgrid(1,1:Nn,1,1:Mn);
tmp = bsxfun(@plus,M1m.*t(N1m)/T,M2m.*t(N2m)/T);
tmp = bsxfun(@times,fun,exp(-1i*2*pi*tmp));
new = coef * permute(sum(sum(tmp,1),2),[3,4,1,2]);
toc
% Maximum difference in outputs:
max(max(abs(old-new)))

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Loops and Conditional Statements 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by