python - python中的矩阵 split 和乘法

Question

我有这 2 个矩阵:第一个 [P,1] 和第二个 [N*P,2]，在 Python 中。

我想将第一个与第二个的 [P,2] 子矩阵相乘，N 次(没有 for 循环)。

我举个例子:

a=[1,2,3]
b=[[1,2],[3,4],[5,6],[7,8],[9,10],[11,12]]

因此，在这种情况下，P=3 N=2。结果应该是一个矩阵 [N,2]。在示例中:

res=[[22,28],[58,64]]

我尝试过使用 reshape(P,N*2)，但我认为它不值得。有什么建议吗？

Answer 1

根据你的订单，你想要的是

np.sum(a[None,:,None]*b.reshape(2,3,2), axis=1)

您需要在中间空间中重新转置数据，因为它们没有像我们预期的那样交错。