Skip to content

Instantly share code, notes, and snippets.

@xnorpx
xnorpx / DFT_ANN.py
Created September 16, 2020 23:26 — forked from endolith/DFT_ANN.py
Training neural network to implement discrete Fourier transform (DFT/FFT)
"""
Train a neural network to implement the discrete Fourier transform
"""
from keras.models import Sequential
from keras.layers import Dense
import numpy as np
import matplotlib.pyplot as plt
N = 32
batch = 10000