mlpack  master
output_parameter_visitor.hpp
Go to the documentation of this file.
1 
13 #ifndef MLPACK_METHODS_ANN_VISITOR_OUTPUT_PARAMETER_VISITOR_HPP
14 #define MLPACK_METHODS_ANN_VISITOR_OUTPUT_PARAMETER_VISITOR_HPP
15 
18 
19 #include <boost/variant.hpp>
20 
21 namespace mlpack {
22 namespace ann {
23 
27 class OutputParameterVisitor : public boost::static_visitor<arma::mat&>
28 {
29  public:
31  template<typename LayerType>
32  arma::mat& operator()(LayerType* layer) const;
33 };
34 
35 } // namespace ann
36 } // namespace mlpack
37 
38 // Include implementation.
39 #include "output_parameter_visitor_impl.hpp"
40 
41 #endif
Linear algebra utility functions, generally performed on matrices or vectors.
Definition: binarize.hpp:18
OutputParameterVisitor exposes the output parameter of the given module.
arma::mat & operator()(LayerType *layer) const
Return the output parameter set.