你將如何從這個MATLAB代碼

Question

消除環路鑑於我們已：

• x是與大小的二維矩陣[NUMSAMPLES X numFeatures]
• A是2D方陣大小[numFeatures X numFeatures]
• B是一個一維向量，其具有尺寸[1×numFeatures]

我想沒有循環來計算下面的代碼：（或者在一個方式更快的方式）

out = zeros(1,numSamples); 
for i = 1:numSamples 
    res = sum(repmat(B - x(i,:), numSamples, 1)*A.*(x - repmat(x(i,:), numSamples, 1)), 2).^2; 
    out(i) = var(res); 
end 

如果您對上述更快的改進有其他建議，它也是值得歡迎的。

Answer 1

您可以bsxfun那些件逐件進行矢量化的解決方案 -

P1 = bsxfun(@minus,B,x)*A; 
P2 = bsxfun(@minus,x,permute(x,[3 2 1])); 
out = var(squeeze(sum(bsxfun(@times,P1,P2),2)).^2.'); 

部分矢量方法 -

P = (bsxfun(@minus,B,x)*A).'; %//' 
out = zeros(1,numSamples); 
for i = 1:numSamples 
    out(i) = var((bsxfun(@minus,x,x(i,:))*P(:,i)).^2); 
end