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

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

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

3
docs/source/_rst/_code.rst
View File

@@ -56,6 +56,7 @@ Models
MIONet <models/mionet.rst>
FourierIntegralKernel <models/fourier_kernel.rst>
FNO <models/fno.rst>
AveragingNeuralOperator <models/avno.rst>
Layers
-------------
@@ -67,10 +68,10 @@ Layers
EnhancedLinear layer <layers/enhanced_linear.rst>
Spectral convolution <layers/spectral.rst>
Fourier layers <layers/fourier.rst>
Averaging layer <layers/avno_layer.rst>
Continuous convolution <layers/convolution.rst>
Proper Orthogonal Decomposition <layers/pod.rst>
Periodic Boundary Condition embeddings <layers/embedding.rst>
Equations and Operators
-------------------------

8
docs/source/_rst/layers/avno_layer.rst Normal file
View File

@@ -0,0 +1,8 @@
Averaging layers
====================
.. currentmodule:: pina.model.layers.avno_layer
.. autoclass:: AVNOBlock
:members:
:show-inheritance:
:noindex:

7
docs/source/_rst/models/avno.rst Normal file
View File

@@ -0,0 +1,7 @@
Averaging Neural Operator
==============================
.. currentmodule:: pina.model.avno
.. autoclass:: AveragingNeuralOperator
:members:
:show-inheritance: