我有一个点xy与形状(2,100)的列表。我想采用带有2x2矩阵的点积,如下所示:
g = xy.T@W@xy
这应该导致100个值的向量。我怎么能用Python做到这一点?
我知道它应该产生100个值,因为如果我输入一个2D点,上面的表达效果很好。我怎样才能对上述内容进行矢量化?
我们可以np.einsum -
np.einsum
np.einsum('ij,ik,kj->j',xy,W,xy, optimize=True)