Numpy Fft2 Training