83ecdb0eab
* Adding fourier block 1d/2d/3d * Adding docs to SpectralConvBlock1D/2D/3D and to FourierBlock1D/2D/3D * Adding tests for fourier block
16 lines
446 B
Python
16 lines
446 B
Python
__all__ = [
|
|
'ContinuousConvBlock',
|
|
'ResidualBlock',
|
|
'SpectralConvBlock1D',
|
|
'SpectralConvBlock2D',
|
|
'SpectralConvBlock3D',
|
|
'FourierBlock1D',
|
|
'FourierBlock2D',
|
|
'FourierBlock3D',
|
|
]
|
|
|
|
from .convolution_2d import ContinuousConvBlock
|
|
from .residual import ResidualBlock
|
|
from .spectral import SpectralConvBlock1D, SpectralConvBlock2D, SpectralConvBlock3D
|
|
from .fourier import FourierBlock1D, FourierBlock2D, FourierBlock3D
|