MATLAB Behavior: for
Conventional wisdom for programming MATLAB used to be that using for loops automatically forced a program to suffer from poor performance. Since MATLAB R13 (version 6.5), MATLAB has taken advantage of some innovations that accelerate many for loops so the code has performance on par with either vectorized code or code written in a lower level language such as C or Fortran. Obviously, details matter here. One thing that most people, even at MathWorks (!) don't appreciate is that the for loop has richer behavior than simply looping over single elements at a time. An informal hallway survey near my office found that even among experienced MATLAB programmers, far fewer than 50% knew about this behavior. Time to come clean.
Contents
Exploring the Behavior of 'for'
The MATLAB for statement is both more powerful and more subtle than many people realize because of the way it lets you iterate directly over an array rather than making use of explicit indices or subscripts. Here's an example that displays the logarithms of the positive numbers in a row vector:
x = [1 pi -17 1.3 289 -exp(1) -42]; for pnum = x(x > 0) disp('Iterating') log(pnum) end
Iterating ans = 0 Iterating ans = 1.1447298858494 Iterating ans = 0.262364264467491 Iterating ans = 5.66642668811243
This code is streamlined in comparison to the following, which uses an explicit index ind, with equivalent results (except for the details on the output that I added here).
pnums = x(x > 0); for ind = 1:numel(pnums) disp(['Iterating... Value #',int2str(ind)]) log(pnums(ind)) end
Iterating... Value #1 ans = 0 Iterating... Value #2 ans = 1.1447298858494 Iterating... Value #3 ans = 0.262364264467491 Iterating... Value #4 ans = 5.66642668811243
Behavior of for
To use for as in the first example, you need to understand that for loops do not iterate over the first dimension of an array, but only over dimensions 2 to the maximum array dimension. See this by transposing the input vector (turning it into a column).
xt = x.' for pnum = xt(xt > 0) disp('Iterating') log(pnum) end
xt = 1 3.14159265358979 -17 1.3 289 -2.71828182845905 -42 Iterating ans = 0 1.1447298858494 0.262364264467491 5.66642668811243
We see one iteration, with pnum taking on the 4-by-1 value [1 pi 1.3 289]'. In contrast, the version with the explicit index works the same way whatever the shape of x.
More Examples
This iterates over s.
for s = [1,-2,8,pi,17], disp('Iterating'), disp(s), end
Iterating 1 Iterating -2 Iterating 8 Iterating 3.14159265358979 Iterating 17
This doesn't iterate, but processes the entire column as one entity.
for s = [1,-2,8,pi,17]', disp('Iterating'), disp(s), end
Iterating 1 -2 8 3.14159265358979 17
What about Higher Dimensions?
This iterates over the 2nd and 3rd dimensions of A.
A = (1:12); A = reshape(A,[2 2 3])
A(:,:,1) = 1 3 2 4 A(:,:,2) = 5 7 6 8 A(:,:,3) = 9 11 10 12
for k = A, disp('iterating'), disp(k), end
iterating 1 2 iterating 3 4 iterating 5 6 iterating 7 8 iterating 9 10 iterating 11 12
The documentation makes it clear that for does not iterate over the first dimension. for does iterate over all of the dimensions of A except for the first (row) dimension. You can predict the number of iterations evaluating this.
numel(A,1,':')
ans = 6
Returning to the first example, numel(x(x >= 0),1,':') evaluates to 4 when x is a row vector and 1 when x is a column vector.
Possible Uses
Why use this version of for? Suppose you have a large dataset and the vectorized calculations can't take full advantage of functions such as bsxfun. You may have a case where the memory tradeoff for vectorization is too high. But you don't want to make function calls for each array element since the function call overhead can get high as well. A possibly good compromise is to essentially process the data in chunks, perhaps by 'virtual' columns. That way the function call overhead is more limited as is the memory. To get the best outcome for this approach, you should preallocate the output array and assign into it with proper indexing as you loop.
What Do You Use?
Did you know about this for behavior? Do you ever take advantage of it? What strategies do you use for trading off memory use and function call overhead? Let me know here.
评论
要发表评论,请点击 此处 登录到您的 MathWorks 帐户或创建一个新帐户。