Is there a way to vectorise this vector/matrix product?
1 visualizzazione (ultimi 30 giorni)
Mostra commenti meno recenti
I have a piece of code inside a for loop, which looks like
for k = 1:size(G, 1)
G(k) = real(W(:,k)' * R * W(:,k))
end
G is preallocated (which is how I know the length of it). W and R are 2D matrices (so W(:,k) is a vector). This loop is a real bottleneck in my code, that one line taking 90% of the total time that it takes to run a script which itself calls other functions and files. It's not possible to use a parfor loop as I'm already using one in a wrapper function.
Is there a way to vectorise that loop, in order to reduce the number of times the calculation is done? I'm sure it can be done, but it's not at all obvious to me.
0 Commenti
Risposta accettata
Matt J
il 7 Mag 2021
Modificato: Matt J
il 7 Mag 2021
G=real( sum((R*W).*conj(W),1) );
4 Commenti
Matt J
il 7 Mag 2021
Here is a test of a agreement with random data:
N=5;
W=complex(rand(N),rand(N));
R=complex(rand(N),rand(N));
%%% Original method
G=nan(N,1);
for k = 1:size(G, 1)
G(k) = real(W(:,k)' * R * W(:,k));
end
Gbase=G;
%%% Proposed alternative
G=real( sum((R*W).*conj(W),1) ).';
Discrepancy=norm(G-Gbase,'fro')
Più risposte (1)
Vedere anche
Categorie
Scopri di più su Loops and Conditional Statements in Help Center e File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!