Files
PINA/pina/model/layers/__init__.py
Dario Coscia b7f9694cb9 Spectral Convolution Addition
* Implementing 1D/2D/3D spectral conv
* Implementing tests for 1D/2D/3d spectral conv
2023-11-17 09:51:29 +01:00

12 lines
311 B
Python

__all__ = [
'ContinuousConvBlock',
'ResidualBlock',
'SpectralConvBlock1D',
'SpectralConvBlock2D',
'SpectralConvBlock3D'
]
from .convolution_2d import ContinuousConvBlock
from .residual import ResidualBlock
from .spectral import SpectralConvBlock1D, SpectralConvBlock2D, SpectralConvBlock3D