Files
OpenCL-examples/example05/main.py
Dakota St. Laurent bff9fd6443 - update Example05 to use FFTW to check the answer
- update clFFT installation instructions
2015-08-12 16:34:10 -04:00

14 lines
193 B
Python

import numpy as np
import scipy.fftpack as fft
N = 2048
v = np.arange(N)
v_fft = fft.fft(v)
v_fft_altered = 2*v_fft.real + 4*v_fft.imag*1j
v_final = fft.ifft(v_fft_altered)
print v_final