Fourier and inverse fourier

how can we use Fourier and inverse Fourier transformation using modulus.
suppose we have some equation and we want to solve it using Semi-Implicit Fourier-Spectral Method,

Hi @big4085boss

You can do any sort of manipulation of data through custom nodes which can then get chained in your constraint, so long as auto-diff can work with it. Nodes just add some meta data to torch modules, so what exactly you do inside of there it up to you so long that you symbolically define what the input / output is. Our FNO model uses PyTorch’s FFTs under the hood.

If things are even more complex, you may have to write a custom constraint.