ここから始まります。まず、間違いをおmyびします。
私はいくつかの異なるアプローチを試しました。上限があるべきである-私は、合計に対する制限で混乱ビットがあったのではなく、?私i − 1
編集:いいえ、質問で提供されているとおり、上限は正しいものでした。別の答えが同じコードを使用するようになったため、ここにそのまま残しましたが、修正は簡単です。
最初にループバージョン:
def looped_ver(k, a):
x = np.empty_like(a)
for i in range(x.size):
sm = 0
for j in range(0, i+1):
sm += k[i-j,j] * a[i-j] * a[j]
x[i] = sm
return x
私はそれをnumpyスライスで単一ループにしました:
def vectorized_ver(k, a):
ktr = zeros_like(k)
ar = zeros_like(k)
sz = len(a)
for i in range(sz):
ktr[i,:i+1] = k[::-1].diagonal(-sz+i+1)
a_ = a[:i+1]
ar[i,:i+1] = a_[::-1] * a_
return np.sum(ktr * ar, 1)
場合、明示的なループが1つあるnumpyバージョンは、コンピューター上で約25倍高速です。n = 5000
次に、Cythonバージョンの(読みやすい)ループコードを作成しました。
import numpy as np
import cython
cimport numpy as np
@cython.boundscheck(False)
@cython.wraparound(False)
def cyth_ver(double [:, ::1] k not None,
double [:] a not None):
cdef double[:] x = np.empty_like(a)
cdef double sm
cdef int i, j
for i in range(len(a)):
sm = 0.0
for j in range(i+1):
sm = sm + k[i-j,j] * a[i-j] * a[j]
x[i] = sm
return x
私のラップトップでは、これはループバージョンよりも約200倍高速です(1ループベクトル化バージョンよりも8倍高速)。他の人がもっとうまくできると確信しています。
私はジュリアバージョンでプレイしましたが、(適切にタイミングを合わせれば)Cythonコードに匹敵するように見えました。