من با کتابخانه numpy برای شما نمونه کد قرار دادم.
from __future__ import print_function
import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 2 * np.pi, 30)
wave = np.cos(x)
transformed = np.fft.fft(wave)
plt.plot(transformed)
plt.title('Transformed cosine')
plt.xlabel('Frequency')
plt.ylabel('Amplitude')
plt.grid()
plt.show()
با خود تنسوفلو هم می تونید fft بگیرید باید نسخه 1.3 به بعد را استفاده کنید برای cpu و نسخه های پایین تر برای gpu