neuralop.models
.TFNO3d
- class neuralop.models.TFNO3d(*args, **kwargs)
3D Fourier Neural Operator
For the full list of parameters, see
neuralop.models.FNO
.- Parameters:
- modes_widthint
number of modes to keep in Fourier Layer, along the width
- modes_heightint
number of Fourier modes to keep along the height
- modes_depthint
number of Fourier modes to keep along the depth
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