43f69242ab
* add Averaging Neural Operator with tests * add backward test * minor changes * doc addition --------- Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.local>
19 lines
489 B
Python
19 lines
489 B
Python
__all__ = [
|
|
"FeedForward",
|
|
"ResidualFeedForward",
|
|
"MultiFeedForward",
|
|
"DeepONet",
|
|
"MIONet",
|
|
"FNO",
|
|
"FourierIntegralKernel",
|
|
"KernelNeuralOperator",
|
|
"AveragingNeuralOperator",
|
|
]
|
|
|
|
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
|