Unifying integral kernel NO architectures (#239)

* 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

View File

@@ -48,11 +48,13 @@ Models
:maxdepth: 5
Network <models/network.rst>
KernelNeuralOperator <models/base_no.rst>
FeedForward <models/fnn.rst>
MultiFeedForward <models/multifeedforward.rst>
ResidualFeedForward <models/fnn_residual.rst>
DeepONet <models/deeponet.rst>
MIONet <models/mionet.rst>
FourierIntegralKernel <models/fourier_kernel.rst>
FNO <models/fno.rst>
Layers

View File

@@ -0,0 +1,7 @@
KernelNeuralOperator
=======================
.. currentmodule:: pina.model.base_no
.. autoclass:: KernelNeuralOperator
:members:
:show-inheritance:

View File

@@ -0,0 +1,7 @@
FourierIntegralKernel
=========================
.. currentmodule:: pina.model.fno
.. autoclass:: FourierIntegralKernel
:members:
:show-inheritance: