用矩阵乘法加速 for 循环？

Question

我的程序的一部分包含这段代码：

size2=2500;
gran=3; 

A=ones(size2,size2);

for k=1:gran:(size2-gran)
    for j=1:gran:(size2-gran)

       X=rand*2*pi-pi;

       for h=1:gran
            for l=1:gran

               A(k+l-1,j+h-1) = A(k+l-1,j+h-1) *exp(+1i*X);  %phase in the square gran x gran

            end
        end
    end
end

我的电脑在 0.60 秒内运行此代码，但我想知道是否可以加快此过程。

一种更快的方法是将其编写为矩阵乘法，但为了编写 X，我认为我必须创建一个 for 循环。

有什么办法可以提高这段代码的速度吗？

Answer 1

您的 for 循环需要替换为具有 2x2 重复的随机相位矩阵。可以将随机变量创建为大小为A的矩阵

X = rand(size2/2)*2*pi-pi;
X = kron(X,ones(2));

然后

A = A.*X;