Add Averaging Neural Operator with tests and a tutorial (#230)

* add Averaging Neural Operator with tests
* add backward test
* minor changes
* doc addition

---------

Co-authored-by: Dario Coscia <dariocoscia@Dario-Coscia.local>
This commit is contained in:
guglielmopadula
2024-03-05 12:30:53 +01:00
committed by GitHub
parent b10e02103b
commit 43f69242ab
8 changed files with 254 additions and 1 deletions

View File

@@ -7,6 +7,7 @@ __all__ = [
"FNO",
"FourierIntegralKernel",
"KernelNeuralOperator",
"AveragingNeuralOperator",
]
from .feed_forward import FeedForward, ResidualFeedForward
@@ -14,3 +15,4 @@ 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