MIONet addition and DeepONet modification.

* Create MIONet base class
* DeepONet inherits from MIONet
This commit is contained in:
Dario Coscia
2023-09-13 09:46:52 +02:00
committed by Nicola Demo
parent 603f56d264
commit ba7371f350
3 changed files with 322 additions and 79 deletions

View File

@@ -2,10 +2,11 @@ __all__ = [
'FeedForward',
'MultiFeedForward',
'DeepONet',
'MIONet',
'FNO',
]
from .feed_forward import FeedForward
from .multi_feed_forward import MultiFeedForward
from .deeponet import DeepONet
from .deeponet import DeepONet, MIONet
from .fno import FNO