向量化循環

Question

在MATLAB中，向量化代碼比使用for循環要快。我一直在試圖做到這一點，但我不完全明白這是如何工作的。我希望有人能告訴我如何改進這兩個for循環和一個while循環與改變索引，所以我可以讓我的頭圍繞它。任何幫助將是一個驚人的。

width= 700; 
height= 600; 
fg= zeros(height, width); 
for i= 1: height 
    for j= 1: width 
     fg(i, j) = 0; 
     while ((match== 0)&& (k<= M)) 
      if (w(i, j, rank_ind(k))>= thresh) 
       if (abs(u_diff(i, j, rank_ind(k)))<= D* sd(i, j, rank_ind(k))) 
        fg(i, j)= 0; 
        match= 1; 
       else 
        fg(i, j)= fr_bw(i, j); 
       end 
      end 
      k= k+ 1; 
     end 
    end 
end 

注意w，u_diff，sd，rank_ind和fr_b都陣列

Answer 1

讓我看看我是否理解正確的話：你要的fr_bw值複製到fg只有在u_diff沒有相應的值小於D*sd，並且w高於某個閾值，對不對？

在這種情況下，你可以重寫代碼的方式如下：

%# find where u_diff is larger than D*sd 
%# but not where there's any u_diff that is smaller than D*sd 
whereToCopy = any(w(:,:,rank_ind) >= thresh & ... 
    abs(u_diff(:,:,rank_ind)) > D*sd(:,:,rank_ind),3) & ... 
    ~any(w(:,:,rank_ind) >= thresh & ... 
    abs(u_diff(:,:,rank_ind)) <= D*sd(:,:,rank_ind),3); 

%# whereToCopy has 1 wherever we need to copy, and 0 elsewhere 
%# thus, create fg from fr_bw directly by multiplication 
fg = double(fr_bw) .* double(whereToCopy);