I would like to play with fast Fourier transform and TF2. I have written the following code in google Colab, but I am doing something wrong. So this computes: