Files
PINA/pina/model/__init__.py
Dario Coscia 766856494a Low Rank Neural Operator (#270)
* add the Low Rank Neural Operator as Model
* add the Low Rank Layer as Layer
* adding tests
* adding doc 

---------

Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.local>
Co-authored-by: Nicola Demo <demo.nicola@gmail.com>
2024-04-02 10:24:23 +02:00

21 lines
556 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