FedeMiorelli

# -*- coding: utf-8 -*-
"""
Created on 2019-08-22 09:37:36

@author: fmiorell
"""

#  This script registers the "turbo" colormap to matplotlib, and the reversed version as "turbo_r"
#  Reference:  https://ai.googleblog.com/2019/08/turbo-improved-rainbow-colormap-for.html

endolith

"""
Train a neural network to implement the discrete Fourier transform
"""
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense
import numpy as np
import matplotlib.pyplot as plt

N = 32
batch = 10000