沿一個特定維度乘以矩陣

Question

題

我正在尋找一種沿一個特定維度將兩個數組相乘的優雅方法。

類似問題

在官方的matlab論壇上已經存在類似的問題 ，但是該主題已過時（2004）。

例

M1是[6x4x4]矩陣，而M2是[6x1]矩陣，我想將M1與M2沿着M1的第3維相乘（逐個元素）以獲得矩陣M [6x4x4]

等效於：

M1 = rand(6,4,4);
M2 = rand(6,1);

for ii = 1:size(M1,2)
   for jj = 1:size(M1,3)
      M(:,ii,jj) = M1(:,ii,jj).*M2;
   end
end

視覺示例

您知道這樣做的好方法嗎？ （無循環，1或2行解決方案，...）

Answer 1

如果我正確地解釋了您的問題，那么您希望在M1每個空間位置取每個時間切片（即1 x 1 xn ），然后將其與元素大小為nx 1的向量M2逐元素相乘。 bsxfun和permute是非常適合的情況：

M = bsxfun(@times, M1, permute(M2, [2 3 1]));