这个for循环可以根据速度进行矢量化吗

Question

我需要找出H_A中每个元素与H_B中每个元素的差异，这是我的方法

H_A=reshape(H_A,1,[]);
H_B=reshape(H_B,1,[]);
for i=1:SIZE^2
D(i,:)=((H_A(i)-H_B)).^2;
end

这可以针对速度进行矢量化吗。

Answer 1

尝试bsxfun。

H_A = reshape(H_A, [], 1);
H_B = reshape(H_B, 1, []);
D = bsxfun(@minus, H_A, H_B).^2;

Answer 2

您可以使用广播：

H_A = reshape(H_A, [], 1);
H_B = reshape(H_B, 1, []);
(H_A-H_B).^2;

可能是最快的选择。还有pdist2，它允许您计算不同的距离度量：

H_A = reshape(H_A, [], 1);
H_B = reshape(H_B, [],1);
pdist2(H_A,H_B,'squaredeuclidean')