“矩阵明智”乘法

Question

我正在使用形状为 Mx3x3 的线性应用程序。我想将它应用到 Mx3x1 的向量中。并得到最终结果Mx3x1。

使用迭代 M 的循环非常简单，但可能效率低下。我怎样才能以更有效的方式解决它？ 谢谢

Answer 1

一般来说，您可以在 Python 中使用 NumPy 进行线性代数，例如在本例中 numpy.matmul

>>> import numpy as np
>>> a = np.arange(9).reshape((3,3))
>>> a
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])
>>> b = np.arange(3)
>>> b
array([0, 1, 2])
>>> np.matmul(a,b)
array([ 5, 14, 23])