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