use LabelTensor, fix minor, docs

2022-03-29 18:05:26 +02:00
parent 12f4084d7f
commit 6b001c6c53
19 changed files with 370 additions and 322 deletions
--- a/pina/model/init.py
+++ b/pina/model/init.py
@@ -1,7 +1,9 @@
 __all__ = [
    'FeedForward',
    'MultiFeedForward'
+    'DeepONet',
 ]

 from .feed_forward import FeedForward
 from .multi_feed_forward import MultiFeedForward
+from .deeponet import DeepONet