多個向量對的Numpy和點積：如何完成？

Question

我想獲得N個向量對（a_vec [i，：]，b_vec [i，：]）的dot product。

a_vec已成形[N, 3],bvec具有相同的形狀（N 3D矢量）。

我知道它可以通過numpy.dot函數在循環中輕鬆完成。但是不可能以某種方式簡單快速地完成它？

Answer 1

np.einsum會做到這一點：

np.einsum('ij,ij->i', a_vec, b_vec)