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/docs/source/_rst/code.rst
+++ b/docs/source/_rst/code.rst
@@ -5,3 +5,6 @@ Code Documentation
    :maxdepth: 3

    LabelTensor <label_tensor.rst>
+    FeedForward <fnn.rst>
+    DeepONet <deeponet.rst>
+    PINN <pinn.rst>
--- a/docs/source/_rst/deeponet.rst
+++ b/docs/source/_rst/deeponet.rst
@@ -0,0 +1,12 @@
+DeepONet
+===========
+.. currentmodule:: pina.model.deeponet
+
+.. automodule:: pina.model.deeponet
+
+.. autoclass:: DeepONet
+    :members:
+    :private-members:
+    :undoc-members:
+    :show-inheritance:
+    :noindex:
--- a/docs/source/_rst/fnn.rst
+++ b/docs/source/_rst/fnn.rst
@@ -0,0 +1,12 @@
+FeedForward
+===========
+.. currentmodule:: pina.model.feed_forward
+
+.. automodule:: pina.model.feed_forward
+
+.. autoclass:: FeedForward
+    :members:
+    :private-members:
+    :undoc-members:
+    :show-inheritance:
+    :noindex:
--- a/docs/source/_rst/pinn.rst
+++ b/docs/source/_rst/pinn.rst
@@ -0,0 +1,12 @@
+PINN
+====
+.. currentmodule:: pina.pinn
+
+.. automodule:: pina.pinn
+
+.. autoclass:: PINN
+    :members:
+    :private-members:
+    :undoc-members:
+    :show-inheritance:
+    :noindex: