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])

FNO's forward pass
forward(x, output_shape=None, **kwargs)

FNO’s forward pass

  1. Applies optional positional encoding

  2. Sends inputs through a lifting layer to a high-dimensional latent

    space

  3. Applies optional domain padding to high-dimensional intermediate function representation

  4. Applies n_layers Fourier/FNO layers in sequence (SpectralConvolution + skip connections, nonlinearity)

  5. If domain padding was applied, domain padding is removed

  6. Projection of intermediate function representation to the output channels

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