vectorize a block matrix calculation

4 visualizzazioni (ultimi 30 giorni)
Luis Isaac
Luis Isaac il 28 Apr 2017
Modificato: Matt J il 28 Apr 2017
Dear;
I want calculate a big matrix composed of blocks, for that I use the following code:
k=length(a); % Long vector
n=size(xx,1); % 2D or 3D matrix always the same
TP=repmat(zeros(size(xx),k,1); % Big matriz composed by blocks
for i=1:k
TP((i-1)*n+1:i*n,:)=a(i)*xx; % Block i of TP matrix
end
Is there any way to vectorize this code and speed up the calculation Many thanks;

Risposta accettata

Matt J
Matt J il 28 Apr 2017
TP=kron(a(:),xx);
  3 Commenti
Andrei Bobrov
Andrei Bobrov il 28 Apr 2017
TP = reshape(bsxfun(@plus,xx',reshape(a,1,1,[])),size(x,2),[])'
Matt J
Matt J il 28 Apr 2017
Modificato: Matt J il 28 Apr 2017
You could also use my attached tensorfun() utility.
TP=tensorfun(@plus,a(:),xx);
It allows you to do similar things for any block structure and any bi-operand operation (@plus, @minus, etc...) supported by bsxfun.

Accedi per commentare.

Più risposte (0)

Tag

Community Treasure Hunt

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

Start Hunting!

Translated by