用bsxfun批次外挂产品

Question

假设我有两个向量，我想做一个外积。我可以用：

A=x*y';

或者我可以像这样使用bsxfun：

bsxfun(@times,x,y')

但我想分批处理外部产品。我有两个矩阵，每个矩阵都包含p向量：

n=1000;  p=6;
A=rand(n,p);
D=rand(n,p);

我要计算所有的外积，并把结果加起来如下：

AA=zeros(n,n);

for j=1:p
    AA = AA + A(:,j) * D(:,j).';
end

我想做得更有效率，但我不知道怎么做。

Answer 1

只需将两个矩阵相乘：

n=1000;  p=6;
A=rand(n,p);
D=rand(n,p);

way1=zeros(n,n);

for j=1:p
    way1 = way1 + A(:,j) * D(:,j).';
end

way2 = A * D.';

any(way1(:) ~= way2(:))

Answer 2

毕竟，您可以使用bsxfun：

C = bsxfun(@times,A,permute(D,[3 2 1]));
result = sum(C,2);

第一行计算A和D之间的所有外部产品，第二行根据请求将结果合并在一起。

经测试如下：

kk = 1e3;
times1 = zeros(kk,1);
n=1000;  p=6;
A=rand(n,p);
D=rand(n,p);
for ii = 1:kk
    tic
    C = bsxfun(@times,A,permute(D.',[3 1 2]));
    result = sum(C,2);
    times1(ii) = toc;
end
mean(times1)

bsxfun是0.0456s，直接乘法是0.0075s，虽然我怀疑我的解决方案实际上做了6次，因为置换，而不是在直接乘法中。因此，每外积bsxfun取0.0076s，几乎相等。