#data generation 512 size import tensorflow as tf from tensorflow.keras.layers import Input import numpy as np import matplotlib.pyplot as plt import math def ge