Fourier Neural Operators for Pytorch.
Single Operator consists of 4 Fourier layers (each layer comprising of SpectralConv and Conv layers).
The overall network:
Lift the input to the desire channel dimension by self.fc0 linear layer.
4 layers of the integral operators u' = (W + K)(u).
W defined by self.w; K defined by self.conv.
Project from the channel space to the output space by self.fc1 and self.fc2 linear layers.
Fourier Neural Operators for Pytorch. Single Operator consists of 4 Fourier layers (each layer comprising of SpectralConv and Conv layers). The overall network: