MATLAB：与向量相乘的 3 维矩阵

Question

我有A 矩阵，即16x16x155460。我有一个B 向量，它是12955x1。我想将每个1:16x1:16x1+12*n:12+12*n 与B(n) 的元素相乘。所以我的目标是根据B找到A的加权和。我这样做的方法如下（I don't want to use for-loop 和我的方法给出了错误的答案，我无法获得连续的 1:12 向量）：

B = repmat(B,[1 16 16]);
B = permute(B,[2 3 1]);
B = repmat(B,[1 1 12]);
result = B.*(A);

作为一个小例子n=2：

A(:,:,1)=[1 2; 3 4]
A(:,:,2)=[1 2; 3 4]
A(:,:,3)=[1 2; 3 4]
A(:,:,4)=[1 2; 3 4]
B      = [2,3]

结果是：

result(:,:,1)=A(:,:,1)*B(1);
result(:,:,2)=A(:,:,2)*B(1);
result(:,:,3)=A(:,:,1)*B(2);
result(:,:,4)=A(:,:,2)*B(2);

Answer 1

如果我理解正确的话，你可以使用bsxfun、permute和reshape这三个强大的三重奏来解决它，就像这样-

[M,N,R] = size(A);
mult_out = bsxfun(@times,reshape(A,M,N,numel(B),[]),permute(B(:),[4 3 1 2]))
out = reshape(mult_out,M,N,[])