Matlab嵌套for循环进入向量化

Question

如果我有一个带有嵌套for 循环的for 循环，使用一个for 循环将值输入到方阵中，循环通过我的Y 轴，另一个循环通过我的X 轴

如何将其更改为矢量化以提高效率

g=zeros(f);

%y loop
for u = 1:f
    %x loop
    for v = 1:f
        g(u,v) = exp(-(((u-u0)^2)+((v-u0)^2))/(2*(sigma^2)))  
    end
end

Answer 1

使用meshgrid

 [u v] = meshgrid(1:f);
 g = exp(-(((u-u0).^2)+((v-u0).^2))/(2*(sigma^2))) ;

这假设sigma、u0 等是标量...

Answer 2

您可以用以下内容替换您的 for 循环：

u = ((1:f) - u0).^2;
v = ((1:f) - v0).^2;
g = exp(-bsxfun(@plus,u',v)/(2*sigma^2));

编辑：

在最新版本的 MATLAB（R2016b 和更高版本）中，您可以在不使用 bsxfun 的情况下计算此值，因为 MATLAB 执行了新的隐式扩展：

u = ((1:f) - u0).^2;
v = ((1:f) - v0).^2;
g = exp((u'+v)/(2*sigma^2));

这应该in theory 和bsxfun 一样快，如果不快的话。