mlpack
forward_visitor.hpp
Go to the documentation of this file.
1 
13 #ifndef MLPACK_METHODS_ANN_VISITOR_FORWARD_VISITOR_HPP
14 #define MLPACK_METHODS_ANN_VISITOR_FORWARD_VISITOR_HPP
15 
16 #include <mlpack/methods/ann/layer/layer_traits.hpp>
17 #include <mlpack/methods/ann/layer/layer_types.hpp>
18 
19 #include <boost/variant.hpp>
20 
21 namespace mlpack {
22 namespace ann {
23 
28 class ForwardVisitor : public boost::static_visitor<void>
29 {
30  public:
32  ForwardVisitor(arma::mat&& input, arma::mat&& output);
33 
35  template<typename LayerType>
36  void operator()(LayerType* layer) const;
37 
38  private:
40  arma::mat&& input;
41 
43  arma::mat&& output;
44 };
45 
46 } // namespace ann
47 } // namespace mlpack
48 
49 // Include implementation.
50 #include "forward_visitor_impl.hpp"
51 
52 #endif
layer_traits.hpp
mlpack
.hpp
Definition: add_to_po.hpp:21
mlpack::ann::ForwardVisitor
ForwardVisitor executes the Forward() function given the input and output parameter.
Definition: forward_visitor.hpp:28
mlpack::ann::ForwardVisitor::operator()
void operator()(LayerType *layer) const
Execute the Foward() function.
mlpack::ann::ForwardVisitor::ForwardVisitor
ForwardVisitor(arma::mat &&input, arma::mat &&output)
Execute the Foward() function given the input and output parameter.
layer_types.hpp