不同维度的矩阵乘法

Question

我需要向量化下面的循环

a=rand(m,n,k)
b=rand(n,k)
c=zeros(m,k)
for ik=1:k
  c(:,ik)=a(:,:,ik)*b(:,ik)
end

我找不到任何执行此操作的 matlab 函数，我认为 bsxfun @multiply 做了一些不同的事情。你能帮忙做这件事吗？

Answer 1

我认为您可以按如下方式使用bsxfun（现在无法对此进行测试 - 如果这给您带来麻烦，请告诉我）：

c = squeeze(sum(bsxfun(@times, a, b), 2));

bsxfun 将扩展矩阵 b 然后进行逐个元素的乘法。第二维上的求和运算负责“矩阵乘法”方面。您可能需要扩展 b 以获得明确的单例第一维：

c = squeeze(sum(bsxfun(@times, a, reshape(b, 1, n, k)), 2));