Vectorising вложенного цикла

Question

for k=1:n 
    for m=1:n_ 
     K_anv(k,m,:)=exp(-(x(k,:)-z(m,:)).^2./(2*l'.^2))-A(k,:).*A_(m,:)./B; 
    end 
end 

K_anv является 3D-матрицей, представляющей собой ядро, где k является количеством данных и m является числом точек оценки. x и z: n x D и n_ x D (lD x 1).

Могу ли я стереть петлю в этом случае?

Answer 1

Поэтапная векторизация с использованием bsxfun -

%// Vectorize "-(x(k,:)-z(m,:)).^2" & thus "exp(-(x(k,:)-z(m,:)).^2./(2*l'.^2))" 
xz = -bsxfun(@minus,permute(x,[1 3 2]),permute(z,[3 1 2])).^2 %//' 
parte1 = exp(bsxfun(@rdivide,xz,permute(2*l'.^2,[1 3 2]))) %//' 

%// Vectorize "A(k,:).*A_(m,:)" & thus "A(k,:).*A_(m,:)./B" 
AA_ = bsxfun(@times,permute(A,[1 3 2]),permute(A_,[3 1 2])) 
parte2 = bsxfun(@rdivide,AA_,permute(B,[1 3 2])) 

%// Finally subtract parte2 from parte1 to get K_anv equivalent 
K_anv_out = parte1 - parte2;