Skip to content

Commit 6e1da6b

Browse files
committed
revise variable name
1 parent 950cddf commit 6e1da6b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

sdp/numpy_fft_sdp.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,13 +9,13 @@ def setup(Q, T):
99
def sliding_dot_product(Q, T, order="F"):
1010
n = len(T)
1111
m = len(Q)
12-
shape = next_fast_len(n, real=True)
12+
next_fast_n = next_fast_len(n, real=True)
1313

14-
tmp = np.empty((2, shape), order=order)
14+
tmp = np.empty((2, next_fast_n), order=order)
1515
tmp[0, :m] = Q[::-1]
1616
tmp[0, m:] = 0.0
1717
tmp[1, :n] = T
1818
tmp[1, n:] = 0.0
1919
fft_2d = np.fft.rfft(tmp, axis=-1)
2020

21-
return np.fft.irfft(np.multiply(fft_2d[0], fft_2d[1]), n=shape)[m - 1 : n]
21+
return np.fft.irfft(np.multiply(fft_2d[0], fft_2d[1]), n=next_fast_n)[m - 1 : n]

0 commit comments

Comments
 (0)