如何在 Matlab 中矢量化双重依赖循环

Question

我有一个循环函数，内部循环取决于外部循环的值。

for jj = 1:500
     for ii = jj:500
         Gamma(ii,jj) =mod( ii-jj, 255);
     end
end

我正在寻找使代码最快的方法：矢量化或bsxfun。现在，我正在使用矢量化方式，但它可能不是最佳的。我问这个问题是为了找到一个更好的解决方案，或者至少比我的方式更好。

[iiValues, jjValues] = meshgrid(1:500, 1:500); 
mask = iiValues >= jjValues;  % ii >= jj
ii= iiValues(mask);    
jj= jjValues(mask);
Gamma(ii,jj)=mod(ii-jj,255) % I am not sure about the line

谢谢

Answer 1

使用 bsxfun 和 tril ：

Gamma = mod(tril(bsxfun(@minus, (1:500).', 1:500)), 255);
%this 1:500 is for the inner loop---^        ^---This 1:500 is for the outer loop

或者使用相同的方法进行隐式扩展（适用于 MATLAB R2016b 及更高版本）：

Gamma = mod(tril((1:500).'-(1:500)), 255);