vectorizing four nested for loops
2 views (last 30 days)
Show older comments
If I have a one-dimensional array A of size n and I need to create a new two-dimensional array R from the elements of array A, I can do it using two nested loops:
for i = 1:n
for j = 1:n
R(i,j) = A(i) + A(j);
end
end
Or I can use vectorization:
[i, j]=ndgrid(1:n);
R = A(i)+A(j);
If array A becomes a n by n matrix and I need to create a four dimensional array R I can still use nested loops:
for i = 1:n
for j = 1:n
for k = 1:n
for l = 1:n
r(i,j, k, l) = a(i, j) + a(k, l);
end
end
end
end
Is there a way to replace these four nested loops and use vectorization? Thanks for your help!
0 Comments
Accepted Answer
Geoff
on 3 Apr 2012
Try this:
[i,j] = ndgrid(1:n*n);
r = reshape(a(i)+a(j), [n n n n]);
4 Comments
Geoff
on 3 Apr 2012
Awwwk!! Um, I dunno. You're talking about chucking in some kind of triangular term. I have trouble thinking about that in 4 dimensions. Remember that brevity should not always be chosen over clarity.
More Answers (1)
See Also
Categories
Find more on Loops and Conditional Statements in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!