Matlab中的高效矩阵乘法

Question

我有两个矩阵，A（N乘K）和B（N乘M）我希望将A和B集中到张量{ {1}}（N由K乘以M）C。我知道如何在python中这样做

C(n,k,m) = A(n,k) * B(n,m)

有谁能告诉我有效做同样事情的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]);