在python中如何计算两个向量数组的点积？

Question

A and B都是具有shape(N,3)的数组。它们各自包含N个载体，使得A[0] = a0 (vector), A[1] = a1...和B[0] = b0, B[1] = b1...

我想计算N对向量an和bn的点积。换言之，我想获得一个数组C，shape(N,1)，这样C[i] = np.dot(A[i],B[i]).在python中，最有效的方法是什么（例如使用矢量化代码）？

Answer 1

你可以执行元素相乘，然后沿着第二个轴求和，就像-

C = (A*B).sum(1)

这些乘法和求和操作可以用^{}一次性实现，就像-

C = np.einsum('ij,ij->i',A,B)