Azzera filtri
Azzera filtri

Elemental matrix multiplication [NxN] and [1xn] into [NxNxn]

3 visualizzazioni (ultimi 30 giorni)
Hi,
I want to multiply every element in A[NxN] by every value in B[1xn] to get C[NxNxn]
Current code:
for i = 1:n
c(:,:,i) = b(i) * a(:,:)
end
Is it possible to do this without the for loop to make it run faster?
Many thanks,

Risposta accettata

Azzi Abdelmalek
Azzi Abdelmalek il 31 Mar 2014
Modificato: Azzi Abdelmalek il 31 Mar 2014
a=rand(4,3)
b=rand(1,6)
[n,m]=size(a)
p=numel(b);
c=reshape(bsxfun(@times, repmat(a(:)',numel(b),1),b')',n,m,p)
  2 Commenti
John D'Errico
John D'Errico il 31 Mar 2014
Modificato: John D'Errico il 31 Mar 2014
That seems overly difficult the way you did it. Cleaner is just to reshape B first, THEN use bsxfun. No need at all to use repmat!
C = bsxfun(@times,A,reshape(B,[1,1,numel(B)]));
The idea is that BSXFUN will implicitly replicate as necessary those dimensions that are singleton (i.e., have a 1 in them.)
William
William il 31 Mar 2014
That's great, faster now. Many thanks

Accedi per commentare.

Più risposte (0)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by