12#ifndef MLPACK_METHODS_ANN_LAYER_MINIBATCH_DISCRIMINATION_HPP
13#define MLPACK_METHODS_ANN_LAYER_MINIBATCH_DISCRIMINATION_HPP
50 typename InputDataType = arma::mat,
51 typename OutputDataType = arma::mat
69 const size_t features);
84 void Forward(
const arma::Mat<eT>& input, arma::Mat<eT>& output);
97 const arma::Mat<eT>& gy,
107 template<
typename eT>
109 const arma::Mat<eT>& ,
110 arma::Mat<eT>& gradient);
128 OutputDataType
const&
Delta()
const {
return delta; }
130 OutputDataType&
Delta() {
return delta; }
133 OutputDataType
const&
Gradient()
const {
return gradient; }
140 template<
typename Archive>
154 OutputDataType weights;
157 OutputDataType weight;
166 arma::cube distances;
169 OutputDataType delta;
172 OutputDataType deltaTemp;
175 OutputDataType gradient;
178 InputDataType inputParameter;
181 OutputDataType outputParameter;
188#include "minibatch_discrimination_impl.hpp"
Implementation of the MiniBatchDiscrimination layer.
OutputDataType const & Delta() const
Get the delta.
OutputDataType const & Parameters() const
Get the parameters.
void Reset()
Reset the layer parameter.
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...
InputDataType & InputParameter()
Modify the input parameter.
OutputDataType const & OutputParameter() const
Get the output parameter.
void Gradient(const arma::Mat< eT > &input, const arma::Mat< eT > &, arma::Mat< eT > &gradient)
Calculate the gradient using the output delta and the input activation.
MiniBatchDiscrimination()
Create the MiniBatchDiscrimination object.
OutputDataType const & Gradient() const
Get the gradient.
InputDataType const & InputParameter() const
Get the input parameter.
MiniBatchDiscrimination(const size_t inSize, const size_t outSize, const size_t features)
Create the MiniBatchDiscrimination layer object using the specified number of units.
OutputDataType & Gradient()
Modify the gradient.
void Backward(const arma::Mat< eT > &, 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...
OutputDataType & OutputParameter()
Modify the output parameter.
void serialize(Archive &ar, const unsigned int)
Serialize the layer.
OutputDataType & Parameters()
Modify the parameters.
OutputDataType & Delta()
Modify the delta.
Linear algebra utility functions, generally performed on matrices or vectors.
The core includes that mlpack expects; standard C++ includes and Armadillo.