* Unify integral kernel NO architectures with NeuralKernelOperator * Implement FNO based on NeuralKernelOperator * modify doc for FNO and add for FourierIntegralKernel, NeuralKernelOperator * adding tests --------- Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.local> Co-authored-by: Dario Coscia <dariocoscia@dhcp-015.eduroam.sissa.it>
17 lines
415 B
Python
17 lines
415 B
Python
__all__ = [
|
|
'FeedForward',
|
|
'ResidualFeedForward',
|
|
'MultiFeedForward',
|
|
'DeepONet',
|
|
'MIONet',
|
|
'FNO',
|
|
'FourierIntegralKernel',
|
|
'KernelNeuralOperator'
|
|
]
|
|
|
|
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
|