多个向量对的 Numpy 和点积:如何完成?

Numpy and dot products of multiple vector pairs: how can it be done?

我想获得 dot product 个向量对 (a_vec[i, :], b_vec[i, :])。

a_vec 的形状为 [N, 3]bvec 的形状相同(N 个 3D 向量)。

我知道可以通过 numpy.dot 函数轻松循环完成。但是不能以某种方式更简单、更快地完成吗?

np.einsum 会做:

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