Matlab中的高效矩阵乘法

Question

我有两个矩阵，A（N x K）和B（N x M），我想将A和B集中到张量C（N x K x M ) 其中C(n,k,m) = A(n,k) * B(n,m)。我知道如何在 python 中做到这一点，比如

C = B[:,numpy.newaxis,:] * A[:,:,numpy.newaxis]

谁能告诉我高效地做同样事情的matlab代码吗？

Answer 1

利用bsxfun 的隐式扩展功能。使用permute 将您的B 用作Nx1xM 矩阵：

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

从 MATLAB R2016b 开始，您可以通过这种方式获得相同的结果：

C = A * permute(B, [1, 3, 2]);