766856494a
* add the Low Rank Neural Operator as Model * add the Low Rank Layer as Layer * adding tests * adding doc --------- Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.local> Co-authored-by: Nicola Demo <demo.nicola@gmail.com>
30 lines
823 B
Python
30 lines
823 B
Python
__all__ = [
|
|
"ContinuousConvBlock",
|
|
"ResidualBlock",
|
|
"EnhancedLinear",
|
|
"SpectralConvBlock1D",
|
|
"SpectralConvBlock2D",
|
|
"SpectralConvBlock3D",
|
|
"FourierBlock1D",
|
|
"FourierBlock2D",
|
|
"FourierBlock3D",
|
|
"PODBlock",
|
|
"PeriodicBoundaryEmbedding",
|
|
"AVNOBlock",
|
|
"LowRankBlock",
|
|
"AdaptiveActivationFunction",
|
|
]
|
|
|
|
from .convolution_2d import ContinuousConvBlock
|
|
from .residual import ResidualBlock, EnhancedLinear
|
|
from .spectral import (
|
|
SpectralConvBlock1D,
|
|
SpectralConvBlock2D,
|
|
SpectralConvBlock3D,
|
|
)
|
|
from .fourier import FourierBlock1D, FourierBlock2D, FourierBlock3D
|
|
from .pod import PODBlock
|
|
from .embedding import PeriodicBoundaryEmbedding
|
|
from .avno_layer import AVNOBlock
|
|
from .lowrank_layer import LowRankBlock
|
|
from .adaptive_func import AdaptiveActivationFunction |