多个向量对的 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)
我想获得 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)