distance.m

来自「流形学习中的重要方法MVU的源代码」· M 代码 · 共 49 行

function dist=distance(X,x);
% dist=distance(X,x)
%
% computes the pairwise squared distance matrix between any column vectors in X and
% in x
%
% INPUT:
%
% X     dxN matrix consisting of N column vectors
% x     dxn matrix consisting of n column vectors
%
% OUTPUT:
%
% dist  Nxn matrix 
%
% Example:
% Dist=distance(X,X);
% is equivalent to
% Dist=distance(X);
%
[D,N] = size(X);
     
try 
 if(nargin>=2)
    % PAIRWISE DISTANCES
  [D,n] = size(x);
  X2 = sum(X.^2,1);
  x2 = sum(x.^2,1);
  dotProd = X'*x; 
 dist = repmat(x2,N,1)+repmat(X2',1,n)-2*dotProd;
 else    
  [D,N] = size(X);
  X2 = repmat(sum(X.^2,1),N,1);
  dist = X2+X2'-2*X'*X;    
 end;
 
catch
  dist=zeros(N);
  tic;
  fprintf('Not enough Memory\nDoing it by hand ...\n');
  for i=1:N-1
    j=(i+1):N;
    dist(i,j)= repmat(X(:,i)'*X(:,i),1,length(j))+sum(X(:,j).^2)-2.*(X(:,j)'*X(:,i))';
    dist(j,i)=dist(i,j)';    
  end;
  fprintf('\n');
end;