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

问题描述 投票:1回答:1

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

g = xy.T@W@xy

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

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

python numpy vectorization
1个回答
2
投票

我们可以np.einsum -

np.einsum('ij,ik,kj->j',xy,W,xy, optimize=True)
© www.soinside.com 2019 - 2024. All rights reserved.