19 lines
525 B
Python
19 lines
525 B
Python
__all__ = [
|
|
'ContinuousConvBlock',
|
|
'ResidualBlock',
|
|
'EnhancedLinear',
|
|
'SpectralConvBlock1D',
|
|
'SpectralConvBlock2D',
|
|
'SpectralConvBlock3D',
|
|
'FourierBlock1D',
|
|
'FourierBlock2D',
|
|
'FourierBlock3D',
|
|
'PODLayer'
|
|
]
|
|
|
|
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 PODLayer
|