Vectorial output

pina/model/feed_forward.py

@@ -52,6 +52,8 @@ class FeedForward(torch.nn.Module):
     def forward(self, x):
         """
         """
+
+        x = x[self.input_variables]
         nf = len(self.extra_features)
         if nf == 0:
             return LabelTensor(self.model(x.tensor), self.output_variables)