行列のペアワイズドット行列

私たちは現在、pythonプロジェクトに取り組んでおり、パフォーマンスの制約のためにベクトル化する必要があります。我々は次の計算に終わる：形状が(20,6)の2つのnumpy配列を持ち、行の対の内積を計算したい、つまり、最後に(20,1)の行列を得るべきである。各行はそれぞれのベクトルによって得られたスカラーであるドットの乗算。行列のペアワイズドット行列

出典

2016-12-25 Sleik

はい、と何をあなたの質問ですか？あなたがこれをやろうとしてあなたの試みでどんな問題を経験していますか？ –

あなたは賢明な2つの配列要素を乗算してから行で合計を行い、その後、あなたはそれぞれの要素が2つの元の配列の行からの内積である配列持つことができます。

a = np.array([[1,2], [3,4]]) 
b = np.array([[3,4], [2,1]]) 

(a * b).sum(axis=1) 
# array([11, 10])

出典

2016-12-25 16:20:23 Psidom

それは、ありがとう！ – Sleik

行列のペアワイズドット行列

答えて

関連する問題