neuralop.models.TFNO1d

class neuralop.models.TFNO1d(*args, **kwargs)

1D Fourier Neural Operator

For the full list of parameters, see neuralop.models.FNO.

Parameters:

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