fourier-cupy.py

# Laskee input-matriisin Fast Fourier muunnoksen FFT
# ja tuloksesta käänteismuunnoksen IFFT,
# jolloin pitäisi tulla alkuperäinen input-matriisi
# Juhani Kaukoranta 12.4.2019
import cupy as cp
import time
n = int(input("Anna n*n matriisin dimensio n "))
A = cp.random.random((n,n),dtype=cp.float32)
time0 = time.perf_counter()
X = cp.fft.fftn(A)
Y = cp.fft.ifftn(X)
time1 = time.perf_counter()
print("\n Aikaa FFT-muunnokseen ja käänteismuunnokseen kului ",time1-time0," sekuntia")
#print("input-matriisi A: \n",A)
#print("A:n FFT-muunnos \n",X)
#print("käänteismuunnos IFFT, pitäisi olla alkuperäinen matriisi A \n",Y)