folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Unifying integral kernel NO architectures (#239)

Browse Source 
* 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>
This commit is contained in:
Dario Coscia
2024-02-21 11:15:40 +01:00
committed by GitHub
parent eb1af0b50e
commit e516e779f9
7 changed files with 354 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pina/model/__init__.py
View File

@@ -1,13 +1,16 @@
__all__ = [
"FeedForward",
"ResidualFeedForward",
"MultiFeedForward",
"DeepONet",
"MIONet",
"FNO",
'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
from .fno import FNO, FourierIntegralKernel
from .base_no import KernelNeuralOperator