多维矩阵乘法

Question

我想知道是否可以在不使用 for 循环的情况下执行多维矩阵乘法。给定 N×P 矩阵 A 和 N×M×P 矩阵 B，我想计算 M 维向量 y，按元素定义为

y(j) = sum_(i = 1,...,N) sum_(k = 1,...,P) A(i,k)*B(i,j,k)

Answer 1

您可以将A 线性化为行向量，然后将reshape 和permute 数组B 线性化为矩阵，这样所需的结果就是矩阵乘法：

M = 5;
N = 6;
P = 8;
A = rand(N,P);
B = rand(N,M,P);
result = A(:).'*reshape(permute(B, [1 3 2]), [], M);

---

或reshape 矩阵A 使其尺寸与B 的尺寸对齐，使用bsxfun 与单例展开相乘，然后将两个所需尺寸相加：

result = sum(sum(bsxfun(@times, reshape(A, N, 1, P), B), 1), 3);