Matlab：如何嵌套ndgrid？

Question

假設我有三個向量a,b,c並且想對所有組合應用功能f。因此，我使用[A,B,C] = ndgrid(a,b,c); result = f(A,B,C)並使用逐點評估來實現f，例如， (A.^B).*C。

但是，第二個函數g的單獨評估僅取決於a和b而不是c。因此，G=g(A,B)（在我的示例中爲A.^B）的調用將通過冗餘地調用具有相同參數length(c)次的函數來浪費時間。我可以使用單獨的ndgrid，但我希望g的結果存儲在相同的維度中，如A,B和C，因此我可以稍後調用h(G,A,C)沒有進一步修改。

如何才能實現這一目標，以及如何沒有使用嵌套循環for反而拖慢程序浪費這個冗餘空間？

Answer 1

只需使用的A和B第一面的計算，然後複製結果與功能REPMAT使G相同大小的其他變量：

G = g(A(:,:,1),B(:,:,1));  %# Compute one unique plane of G 
G = repmat(G,[1 1 length(c)]); %# Replicate that plane for a 3-D matrix 

編輯：

如果複製G看起來可能會浪費內存（即如果您處理的是非常大的矩陣），那麼for循環可能是更好的方法。另外，您可以嘗試在計算中使用功能BSXFUN。例如，你的3-d C乘以2-d G，你可以這樣做：

bsxfun(@times,G,C); 

這最終會給出相同的結果：

repmat(G,[1 1 length(c)]).*C; 

將其保存在記憶？我不能確定，因爲我不知道BSXFUN是如何實現的。您必須針對您的特定問題嘗試一下，看看哪種方法可以提高速度或內存使用率。