【发布时间】:2019-07-10 13:39:02
【问题描述】:
我有一个形状为 (2,100) 的点 xy 列表。我想采用 2x2 矩阵的点积如下:
g = xy.T@W@xy
这应该产生一个包含 100 个值的向量。如何用 Python 做到这一点?
我知道它应该产生 100 个值,因为如果我输入一个 2D 点,上面的表达式效果很好。如何对上述内容进行矢量化?
【问题讨论】:
标签: python numpy vectorization
我有一个形状为 (2,100) 的点 xy 列表。我想采用 2x2 矩阵的点积如下:
g = xy.T@W@xy
这应该产生一个包含 100 个值的向量。如何用 Python 做到这一点?
我知道它应该产生 100 个值,因为如果我输入一个 2D 点,上面的表达式效果很好。如何对上述内容进行矢量化?
【问题讨论】:
标签: python numpy vectorization
我们可以np.einsum -
np.einsum('ij,ik,kj->j',xy,W,xy, optimize=True)
【讨论】:
xy with the shape(2,100).。因此,无需更改。