Files
PINA/pina/model/__init__.py
2024-04-02 09:25:30 +01:00

21 lines
557 B
Python

__all__ = [
"FeedForward",
"ResidualFeedForward",
"MultiFeedForward",
"DeepONet",
"MIONet",
"FNO",
"FourierIntegralKernel",
"KernelNeuralOperator",
"AveragingNeuralOperator",
"LowRankNeuralOperator",
]
from .feed_forward import FeedForward, ResidualFeedForward
from .multi_feed_forward import MultiFeedForward
from .deeponet import DeepONet, MIONet
from .fno import FNO, FourierIntegralKernel
from .base_no import KernelNeuralOperator
from .avno import AveragingNeuralOperator
from .lno import LowRankNeuralOperator