Question

我需要帮助优化此循环。 matrix_1是（n x 2）int矩阵，matrix_2是（m x 2），m＆amp; n非常。

index_j = 1;
for index_k = 1:size(Matrix_1,1)
    for index_l = 1:size(Matrix_2,1)
        M2_Index_Dist(index_j,:) = [index_l, sqrt(bsxfun(@plus,sum(Matrix_1(index_k,:).^2,2),sum(Matrix_2(index_l,:).^2,2)')-2*(Matrix_1(index_k,:)*Matrix_2(index_l,:)'))];
        index_j = index_j + 1;
    end
 end

我需要M2_Index_Dist提供一个（(n*m) x 2）矩阵，第一列中的索引为matrix_2，第二列中的距离为。{/ p>

输出示例：

M2_Index_Dist = [ 1, 5.465
                  2, 56.52
                  3, 6.21
                  1, 35.3
                  2, 56.52
                  3, 0
                  1, 43.5
                  2, 9.3
                  3, 236.1
                  1, 8.2
                  2, 56.52
                  3, 5.582]

Answer 1

如果我理解正确，这可以做你想要的：

ind = repmat((1:size(Matrix_2,1)).',size(Matrix_1,1),1); %'// first column: index
d = pdist2(Matrix_2,Matrix_1); %// compute distance between each pair of rows
d = d(:); %// second column: distance
result = [ind d]; %// build result from first column and second column

如您所见，此代码调用pdist2来计算矩阵的每对行之间的距离。默认情况下，此函数使用欧几里德距离。

如果您没有pdist2（这是统计工具箱的一部分），您可以将上面的第2行替换为bsxfun：

d = squeeze(sqrt(sum(bsxfun(@minus,Matrix_2,permute(Matrix_1, [3 2 1])).^2,2)));

Answer 2

以下是如何将bsxfun应用于您的公式（||A-B|| = sqrt(||A||^2 + ||B||^2 - 2*A*B)）：

d = real(sqrt(bsxfun(@plus, dot(Matrix_1,Matrix_1,2), ...
    bsxfun(@minus, dot(Matrix_2,Matrix_2,2).', 2 * Matrix_1*Matrix_2.')))).';

如果更改矩阵的解释，则可以避免最终转置。

注意：real不应该有任何复杂的值，但如果差异很小，可能会导致微小的负数。

编辑：没有dot可能会更快：

d = sqrt(bsxfun(@plus, sum(Matrix_1.*Matrix_1,2), ...
    bsxfun(@minus, sum(Matrix_2.*Matrix_2,2)', 2 * Matrix_1*Matrix_2.'))).';

或只需拨打一次bsxfun：

d = sqrt(bsxfun(@plus, sum(Matrix_1.*Matrix_1,2), sum(Matrix_2.*Matrix_2,2)') ...
    - 2 * Matrix_1*Matrix_2.').';

注意：最后一个操作顺序给出了相同的结果，而不是错误~1e-14。

编辑2：要复制M2_Index_Dist：

II = ndgrid(1:size(Matrix_2,1),1:size(Matrix_2,1));
M2_Index_Dist = [II(:) d(:)];

成对L2距离计算的优化

2 个答案: