performance - 如何在 matlab 中使这个代码段更快？

Question

function [D] = distChiSq( W, X )
%%% find the  Chi2Dist distance between each weight vector and X
% W is nxd
m = size(W,1);  n = size(X,1);
k = size(W,2);
mOnes = ones(1,m); D = zeros(m,n);
for i=1:n
    Xi = X(i,:);  XiRep = Xi( mOnes, : );
    s = XiRep + W;                        /************/
    d = XiRep - W;                        /************/
    D(:,i) = sum( d.^2 ./ (s+eps), 2 );   /************/
end
D = D/2;

这是我的权重矩阵和数据之间的 Chi2 距离计算的一部分。标记行是所有代码中消耗最多的行。有什么方法可以在 MATLAB 中更快地做到这一点？

如果数据是 nxm ，s = nxm, d = nxm并且 D 保持距离。n 是实例数，m 是变量数。

Answer 1

哦，这对bsxfun来说太有趣了：

s = bsxfun( @plus, permute( X, [1 3 2] ), permute( W, [3 1 2] ) ) + eps;
d = bsxfun( @minus, permute( X, [1 3 2] ), permute( W, [3 1 2] ) ).^2;
D = .5*sum( d./s, 3 );

Answer 2

替代 Shai 的出色回答：

F = @(a,b) (a-b).^2 ./ (a+b+eps);
D = sum(bsxfun(F, permute(X, [3 1 2]), permute(W, [1 3 2])), 3)/2;