12 #ifndef MLPACK_METHODS_ANN_LAYER_SELECT_HPP 13 #define MLPACK_METHODS_ANN_LAYER_SELECT_HPP 29 typename InputDataType = arma::mat,
30 typename OutputDataType = arma::mat
41 Select(
const size_t index = 0,
const size_t elements = 0);
51 void Forward(
const arma::Mat<eT>&& input, arma::Mat<eT>&& output);
63 void Backward(
const arma::Mat<eT>&& ,
73 OutputDataType&
Delta()
const {
return delta; }
75 OutputDataType&
Delta() {
return delta; }
80 template<
typename Archive>
81 void serialize(Archive& ar,
const unsigned int );
94 OutputDataType outputParameter;
101 #include "select_impl.hpp" void serialize(Archive &ar, const unsigned int)
Serialize the layer.
void Forward(const arma::Mat< eT > &&input, arma::Mat< eT > &&output)
Ordinary feed forward pass of a neural network, evaluating the function f(x) by propagating the activ...
OutputDataType & Delta() const
Get the delta.
The core includes that mlpack expects; standard C++ includes and Armadillo.
OutputDataType & OutputParameter()
Modify the output parameter.
void Backward(const arma::Mat< eT > &&, arma::Mat< eT > &&gy, arma::Mat< eT > &&g)
Ordinary feed backward pass of a neural network, calculating the function f(x) by propagating x backw...
The select module selects the specified column from a given input matrix.
Select(const size_t index=0, const size_t elements=0)
Create the Select object.
OutputDataType & Delta()
Modify the delta.
OutputDataType & OutputParameter() const
Get the output parameter.