Matlab與四維陣列的每個矢量工作

Question

我有一個4D矩陣的大小，比方說，100x200x3x10其中100x200是一個圖像的大小，3是通道數量（RGB圖像），10是圖像數量。

我想計算每個圖像中每個RGB矢量與其自身的內積。生成的圖像的尺寸應爲100x200x10。我怎樣纔能有效地計算這些產品，可能沒有使用循環？

謝謝。

Answer 1

如果你打電話給矩陣M，這應該工作：

squeeze(dot(M,M,3)) 

的squeeze是因爲MATLAB給出了一個100x200x1x10矩陣作爲結果，並squeeze挖空多餘的尺寸。