version 0.0.1

pina/model/multi_feed_forward.py

@@ -0,0 +1,17 @@
+import torch
+
+from .feed_forward import FeedForward
+
+
+class MultiFeedForward(torch.nn.Module):
+
+    def __init__(self, dff_dict):
+        '''
+        '''
+        super().__init__()
+
+        if not isinstance(dff_dict, dict):
+            raise TypeError
+
+        for name, constructor_args in dff_dict.items():
+            setattr(self, name, FeedForward(**constructor_args))