neuralop.models
.FNO2d
- class neuralop.models.FNO2d(*args, **kwargs)[source]
2D Fourier Neural Operator
For the full list of parameters, see
neuralop.models.FNO
.- Parameters:
- n_modes_widthint
number of modes to keep in Fourier Layer, along the width
- n_modes_heightint
number of Fourier modes to keep along the height