多个矢量对的Numpy和dot产品:怎么做?

Fel*_*lix 2 python numpy matrix scipy

我想得到dot productN个向量对(a_vec [i,:],b_vec [i,:]).

a_vec具有形状[N, 3],bvec具有相同的形状(N个3D矢量).

我知道它可以通过numpy.dot功能在循环中轻松完成.但不能以某种方式更简单,更快地完成吗?

use*_*ica 6

np.einsum 会这样做:

np.einsum('ij,ij->i', a_vec, b_vec)
Run Code Online (Sandbox Code Playgroud)