mlpack 3.4.2
delta_visitor.hpp
Go to the documentation of this file.
1
13#ifndef MLPACK_METHODS_ANN_VISITOR_DELTA_VISITOR_HPP
14#define MLPACK_METHODS_ANN_VISITOR_DELTA_VISITOR_HPP
15
18
19#include <boost/variant.hpp>
20
21namespace mlpack {
22namespace ann {
23
27class DeltaVisitor : public boost::static_visitor<arma::mat&>
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 "delta_visitor_impl.hpp"
42
43#endif
DeltaVisitor exposes the delta parameter of the given module.
arma::mat & operator()(MoreTypes layer) const
arma::mat & operator()(LayerType *layer) const
Return the delta parameter.
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
Linear algebra utility functions, generally performed on matrices or vectors.
Definition: cv.hpp:1