mlpack
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 
16 #include <mlpack/methods/ann/layer/layer_traits.hpp>
17 #include <mlpack/methods/ann/layer/layer_types.hpp>
18 
19 #include
20 
21 namespace mlpack {
22 namespace ann {
23 
27 class OutputParameterVisitor : public boost::static_visitor
28 {
29  public:
31  template<typename LayerType>
32  arma::mat& operator()(LayerType* layer) const;
33 
34  arma::mat& operator()(MoreTypes layer) const;
35 };
36 
37 } // namespace ann
38 } // namespace mlpack
39 
40 // Include implementation.
41 #include "output_parameter_visitor_impl.hpp"
42 
43 #endif
layer_traits.hpp
mlpack
Linear algebra utility functions, generally performed on matrices or vectors.
Definition: add_to_cli11.hpp:21
mlpack::ann::OutputParameterVisitor::operator()
arma::mat & operator()(LayerType *layer) const
Return the output parameter set.
mlpack::ann::OutputParameterVisitor
OutputParameterVisitor exposes the output parameter of the given module.
Definition: output_parameter_visitor.hpp:27
mlpack::ann::MoreTypes
boost::variant< Linear3D< arma::mat, arma::mat, NoRegularizer > *, Glimpse< arma::mat, arma::mat > *, Highway< arma::mat, arma::mat > *, MultiheadAttention< arma::mat, arma::mat, NoRegularizer > *, Recurrent< arma::mat, arma::mat > *, RecurrentAttention< arma::mat, arma::mat > *, ReinforceNormal< arma::mat, arma::mat > *, Reparametrization< arma::mat, arma::mat > *, Select< arma::mat, arma::mat > *, Sequential< arma::mat, arma::mat, false > *, Sequential< arma::mat, arma::mat, true > *, Subview< arma::mat, arma::mat > *, VRClassReward< arma::mat, arma::mat > *, VirtualBatchNorm< arma::mat, arma::mat > *, RBF< arma::mat, arma::mat, GaussianFunction > *, BaseLayer< GaussianFunction, arma::mat, arma::mat > *, PositionalEncoding< arma::mat, arma::mat > *> MoreTypes
Definition: layer_types.hpp:238
layer_types.hpp