重み付き合計matlabをベクトル化する

Question

私は一定の加重和をベクトル化しようとしましたが、それを行う方法を理解できませんでした。私は以下の簡単な作業例を作成しました。ソリューションにはbsxfunやreshape、kroneckerのいずれかの製品が含まれていると思いますが、まだそれを稼働させることはできませんでした。

rng(1); 
N = 200; 
T1 = 5; 
T2 = 7; 

A = rand(N,T1,T2); 
w1 = rand(T1,1); 
w2 = rand(T2,1); 

B = zeros(N,1); 

for i = 1:N 
for j1=1:T1 
for j2=1:T2 
    B(i) = B(i) + w1(j1) * w2(j2) * A(i,j1,j2); 
end 
end 
end 

A = B; 

Answer 1

あなたはこれを達成するためにbsxfun、reshapeとpermuteの組み合わせを使用することができます。

最初にpermuteを使用してNディメンションをAの3番目のディメンションに移動します。次に、w1と転置のw2を乗算して、重みのグリッドを作成します。次に、bsxfunを使用して、このグリッドと各「スライス」との間の要素ごとの乗算（@times）をAとします。次に、3D結果をM×Nに再整理し、第1次元全体にわたって合計することができる。

B = sum(reshape(bsxfun(@times, w1 * w2.', permute(A, [2 3 1])), [], N)).'; 

更新

あなたのための加算を実行するために行列の乗算を使用する簡単な方法は、実際にあります。うわー

B = reshape(A, N, []) * kron(w2, w1);