矢量一個求和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));