Построить матрицу с векторами, полученными в результате индексированных продуктов кронекера без цикла for

Question

Могу ли я оптимизировать следующий код в Matlab, обратив петлю for?

A = []; 
B = randn(4,8); 
C = randn(8,4); 
I = randperm(8,3); 
J = randperm(8,3); 
for i = 1:3 
    A = [A kron(C(J(i),:)',B(:,I(i)))]; 
end 

Answer 1

Да, вы можете, используя третье измерение, сохранить промежуточные результаты и преобразовать их обратно в 2D. Таким образом, вы также можете избежать самого kron, который не самый быстрый.

Matlab R2016a или более поздней версии:

a = C(J,:).' .* permute(B(:,I),[3 2 1]); %// calculation of the product to 3rd dimension 
              %// by implicit expansion 
b = permute(a, [3 1 2]);    %// permuting 
out = reshape(b, [], 3)     %// reshape to desired form 

коротко:

out = reshape(permute(C(J,:).' .* permute(B(:,I),[3 2 1]), [3 1 2]), [], 3) 

Перед Matlab R2016a:

a = bsxfun(@times , C(J,:).', permute(B(:,I),[3 2 1])); %// calculation of 
              %// the product to 3rd dimension(explicit) 
b = permute(a, [3 1 2]);    %// permuting 
out = reshape(b, [], 3)     %// reshape to desired form 

коротко:

out = reshape(permute(bsxfun(@times , C(J,:).', permute(B(:,I),[3 2 1])), [3 1 2]), [], 3) 

Answer 2

Вот Векторизованная версия, использующая kron (не так быстро, как thewaywewalk «ы ответа):

kron(C(J,:).', ones(size(B,1),1)) .* kron(ones(size(B,1),1), B(:,I))