Files
PINA/pina/model/layers/__init__.py
Dario Coscia cddb191fe4 Adaptive Functions (#272)
* adaptive function improvement

---------

Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.local>
2024-04-02 10:11:10 +02:00

29 lines
763 B
Python

__all__ = [
"ContinuousConvBlock",
"ResidualBlock",
"EnhancedLinear",
"SpectralConvBlock1D",
"SpectralConvBlock2D",
"SpectralConvBlock3D",
"FourierBlock1D",
"FourierBlock2D",
"FourierBlock3D",
"PODBlock",
"PeriodicBoundaryEmbedding",
"AVNOBlock",
"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 .adaptive_func import AdaptiveActivationFunction