使用2x2矩阵广播2D点列表的点积

Question

我有一个点xy与形状（2,100）的列表。我想采用带有2x2矩阵的点积，如下所示：

g = xy.T@W@xy

这应该导致100个值的向量。我怎么能用Python做到这一点？

我知道它应该产生100个值，因为如果我输入一个2D点，​​上面的表达效果很好。我怎样才能对上述内容进行矢量化？

Answer 1

我们可以np.einsum -

np.einsum('ij,ik,kj->j',xy,W,xy, optimize=True)