neuralop.models
.TFNO2d
- class neuralop.models.TFNO2d(*args, **kwargs)
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
Methods
forward
(x[, output_shape])TFNO's forward pass
- forward(x, output_shape=None, **kwargs)
TFNO’s forward pass
- Parameters:
- xtensor
input tensor
- output_shape{tuple, tuple list, None}, default is None
Gives the option of specifying the exact output shape for odd shaped inputs. * If None, don’t specify an output shape * If tuple, specifies the output-shape of the last FNO Block * If tuple list, specifies the exact output-shape of each FNO Block