多数の小さな線形システムを解く

Question

私は、Pythonを使って3x3の対称でポピュライティブな多数のシステムを解決する必要があります。 nが大きい場合にはこれまでのところ、私はこれが正しい結果を生成

res = numpy.zeros(n) 
for k, obj in enumerate(data_array): 
    # construct A, rhs, idx from obj 
    res[idx] += numpy.linalg.solve(A, rhs) 

をした、しかし、また、非常に遅いです。 （まあ...ええ。）おそらく3x3は問題のサイズではありませんsolve()を呼び出すことは意味をなさないでしょう。

ヒント

Answer 1

NumPy 1.8以降では、numpy.linalg.solve actually broadcastsです。 numpy.linalg.solve(a, b)の場合、b.ndim == a.ndim - 1の場合、ブロードキャストされた行列ベクトル解を実行します。それ以外の場合は、ブロードキャストされた行列行列解を実行します。 （この判断基準は文書化されていません。私は、ソースを見ていた。）

あなたが効率的にA sおよびrhs秒のスタックを構築することができれば、あなたは一度solveを呼び出すと、Pythonのループを回避することができます。