python - Matlab 中的高效矩阵乘法

Question

我有两个矩阵，A(N乘K)和B(N乘M)，我想集中A和B 转换为张量 C(N x K x M)，其中 C(n,k,m) = A(n,k) * B(n,米)。我知道如何在 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]);