13#ifndef MLPACK_METHODS_ANN_LAYER_SOFTMIN_HPP
14#define MLPACK_METHODS_ANN_LAYER_SOFTMIN_HPP
32 typename InputDataType = arma::mat,
33 typename OutputDataType = arma::mat
50 template<
typename InputType,
typename OutputType>
51 void Forward(
const InputType& input, OutputType& output);
64 const arma::Mat<eT>& gy,
73 InputDataType&
Delta()
const {
return delta; }
75 InputDataType&
Delta() {
return delta; }
80 template<
typename Archive>
88 OutputDataType outputParameter;
95#include "softmin_impl.hpp"
Implementation of the Softmin layer.
void Forward(const InputType &input, OutputType &output)
Ordinary feed forward pass of a neural network, evaluating the function f(x) by propagating the activ...
OutputDataType & OutputParameter() const
Get the output parameter.
Softmin()
Create the Softmin object.
void Backward(const arma::Mat< eT > &input, const 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...
InputDataType & Delta()
Modify the delta.
void serialize(Archive &, const unsigned int)
Serialize the layer.
InputDataType & Delta() const
Get the delta.
OutputDataType & OutputParameter()
Modify the output parameter.
Linear algebra utility functions, generally performed on matrices or vectors.
The core includes that mlpack expects; standard C++ includes and Armadillo.