overload Module::operator() to call Module::forward

examples/perceptron.cpp

@@ -51,10 +51,10 @@ int main()
             af::array out_j = out(af::span, j);
 
             // Forward propagation
-            result = perceptron.forward(nn::input(in_j));
+            result = perceptron(nn::input(in_j));
 
             // Calculate loss
-            l = loss.forward(result, nn::noGrad(out_j));
+            l = loss(result, nn::noGrad(out_j));
 
             // Backward propagation
             l.backward();
@@ -71,7 +71,7 @@ int main()
             perceptron.eval();
 
             // Forward propagation
-            result = perceptron.forward(nn::input(in));
+            result = perceptron(nn::input(in));
 
             // Calculate loss
             // TODO: Use loss function

include/af/nn/Modules/Loss.hpp

@@ -23,6 +23,9 @@ namespace af
                                                const autograd::Variable &targets) = 0;
 
             autograd::Variable forward(const autograd::Variable &inputs);
+
+            autograd::Variable operator()(const autograd::Variable &inputs,
+                                          const autograd::Variable &targets);
         };
 
         class MeanSquaredError : public Loss

include/af/nn/Modules/Module.hpp

@@ -42,6 +42,8 @@ namespace af
             void eval();
 
             virtual autograd::Variable forward(const autograd::Variable &input) = 0;
+
+            autograd::Variable operator()(const autograd::Variable &input);
         };
     }
 }

src/nn/Modules/Loss.cpp

@@ -21,6 +21,12 @@ namespace af
             throw af::exception("Loss module requires both inputs and targets");
         }
 
+        autograd::Variable Loss::operator()(const autograd::Variable &inputs,
+                                            const autograd::Variable &targets)
+        {
+            return this->forward(inputs, targets);
+        }
+
         autograd::Variable MeanSquaredError::forward(const autograd::Variable &inputs,
                                                      const autograd::Variable &targets)
         {

src/nn/Modules/Module.cpp

@@ -60,5 +60,10 @@ namespace af
                 parameter.zeroGrad();
             }
         }
+
+        Variable Module::operator()(const Variable &input)
+        {
+            return this->forward(input);
+        }
     }
 }