matlab - 如何在matlab中向量化表达式

Question

我无法对此进行矢量化：

for x=2:i
   for y=2:j
     if(x ~= y)
       Savings(x,y) = Costs(x,1) + Costs(1,y) - Costs(x,y);
     end
   end
end

有人能告诉我我可以提高这段代码的性能吗？谢谢

Answer 1

在一些帮助下bsxfun：

Ix=2:i; 
Iy=2:j;
I = false(i,j);
I(Ix,Iy) = bsxfun(@ne, Ix', Iy);
S = bsxfun(@plus, Costs(Ix,1), Costs(1,Iy)) - Costs(Ix,Iy);
Savings(I) = S(I(Ix,Iy));

Answer 2

你可以像这样矢量化它，我不知道这是否会影响你的表现。你需要自己测试一下。

m=size(Costs, 1);
n=size(Costs, 2);

[Y, X] = meshgrid(2:m, 2:n);

Cx = Costs(:,1);
Cy = Costs(1,:);

S = Cx(X) + Cy(Y) - Costs(2:end,2:end);

S(eye(m-1,n-1)==1) = 0;
Savings = zeros(m,n);
Savings(2:end,2:end) = S;

编辑

顺便说一句，我已经验证了所有三种方法都给出了相同的答案。对于Costs400x400 的大小，运行时间如下：

Elapsed time is 0.00741386 seconds. //My method
Elapsed time is 0.003304 seconds.   //Mohsen's method (after correcting to prevent errors)
Elapsed time is 2.16231 seconds.    //Original Loop

因此，我们的两种方法都提供了显着的提升。但是，如果你只是预先分配Savings你的循环实际上是最快的。对于您的目的来说，这真的太慢了​​吗？