python - 就地矩阵乘法 numpy

Question

我有一个点数组列表，当我迭代它时，我想进行就地矩阵乘法，即我希望将结果存储在同一个矩阵中。

代码本质上是:

for p in p_list:
    # R is a 3x3 matrix
    p[:,:] = np.matmul(R,p)

这段代码没有显示错误，但结果不正确，就好像在数组中执行乘法并按计算结果替换一样，因此它创建了错误的输出矩阵。删除 [:,:] 给出正确的乘法。

1) 为什么会发生这种情况？2) 我使用 [:,:] 的主要原因是确保结果存储回列表 p_list 中。有没有正确的方法来做到这一点(不使用中间变量)？

Answer 1

matmul 接受输出参数

如果p_list是一个ndarray，形状为N, 3，那么您可以在一个matmul<中实现整个乘法:

np.matmul(p_list, R.T, out=p_list)