我有一个形状为 (2,100) 的 xy 点列表。我想采用 2x2 矩阵的点积,如下所示:
g = xy.T@W@xy
这应该产生一个包含 100 个值的向量。我怎样才能用Python做到这一点?
我知道它应该产生 100 个值,因为如果我输入一个 2D 点,上面的表达效果很好。如何向量化上述内容?
最佳答案
我们可以np.einsum
-
np.einsum('ij,ik,kj->j',xy,W,xy, optimize=True)
关于python - 使用 2x2 矩阵广播 2D 点列表的点积,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54725791/