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

@@ -10,6 +10,7 @@ __all__ = [
"FourierBlock3D",
"PODBlock",
"PeriodicBoundaryEmbedding",
"AVNOBlock",
]
from .convolution_2d import ContinuousConvBlock
@@ -22,3 +23,4 @@ from .spectral import (
from .fourier import FourierBlock1D, FourierBlock2D, FourierBlock3D
from .pod import PODBlock
from .embedding import PeriodicBoundaryEmbedding
from .avno_layer import AVNOBlock