vectorize a block matrix calculation

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;

 Respuesta aceptada

Matt J
Matt J el 28 de Abr. de 2017
TP=kron(a(:),xx);

3 comentarios

Many thaks Matt
I wondered, if I would want to calculate the sum, what is the correct operation. I mean:
...
TP((i-1)*n+1:i*n,:)=a(i)+xx; % Block i of TP matrix
...
Thanks in advanced;
TP = reshape(bsxfun(@plus,xx',reshape(a,1,1,[])),size(x,2),[])'
Matt J
Matt J el 28 de Abr. de 2017
Editada: Matt J el 28 de Abr. de 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.

Iniciar sesión para comentar.

Más respuestas (0)

Etiquetas

Preguntada:

el 28 de Abr. de 2017

Editada:

el 28 de Abr. de 2017

Community Treasure Hunt

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

Start Hunting!

Translated by