如何在 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 and 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);

如何在 Matlab 中向量化双依赖循环？

How to vectorize double dependence loop in Matlab?

performance

matlab

vectorization

bsxfun