l2_distance.m

来自「K近邻法的matlab程序」· M 代码 · 共 17 行

function d = l2_distance(X,Y)
% d = l2_distance(X)
% Compute pairwise l2 distances between column vectors in X

if (nargin < 2)
    [D N] = size(X);
    lengths = sum(X.^2,1);
    d = repmat(lengths,[N 1]) + repmat(lengths',[1 N]);
    d = d - 2* X'*X;
    
else 
    XX = sum(X.^2,1);
    YY = sum(Y.^2,1);
    d = repmat(XX', [1 size(Y,2)]) + repmat(YY, [size(X,2) 1]);
    d = d - 2*X'*Y;

end