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

7 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)

Categorie

Scopri di più su Creating and Concatenating Matrices 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!

Translated by