超过3D阵列MATLAB的矢量化

Question

我想知道如何在MATLAB中进行矢量化此代码以避免使用循环：

for i=1:n1
for j=1:n2
A(i,j) = sum(B(:,i,j).*C(:,i,j));
end
end

a是尺寸N1*N2和B，C为3D数组的矩阵。

谢谢。

Answer 1

您应该能够直接做

A = sum(B.*C,1);

或删除第一个维度：

A = squeeze(sum(B.*C,1));