mlpack  3.4.2
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
Public Member Functions | List of all members
Linear3D< InputDataType, OutputDataType, RegularizerType > Class Template Reference

Implementation of the Linear3D layer class. More...

Public Member Functions

 Linear3D ()
 Create the Linear3D object. More...
 
 Linear3D (const size_t inSize, const size_t outSize, RegularizerType regularizer=RegularizerType())
 Create the Linear3D layer object using the specified number of units. More...
 
template<typename eT >
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 backwards trough f. More...
 
OutputDataType const & Bias () const
 Get the bias of the layer. More...
 
OutputDataType & Bias ()
 Modify the bias weights of the layer. More...
 
OutputDataType const & Delta () const
 Get the delta. More...
 
OutputDataType & Delta ()
 Modify the delta. More...
 
template<typename eT >
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 activity forward through f. More...
 
template<typename eT >
void Gradient (const arma::Mat< eT > &input, const arma::Mat< eT > &error, arma::Mat< eT > &gradient)
 
OutputDataType const & Gradient () const
 Get the gradient. More...
 
OutputDataType & Gradient ()
 Modify the gradient. More...
 
InputDataType const & InputParameter () const
 Get the input parameter. More...
 
InputDataType & InputParameter ()
 Modify the input parameter. More...
 
size_t InputSize () const
 Get the input size. More...
 
OutputDataType const & OutputParameter () const
 Get the output parameter. More...
 
OutputDataType & OutputParameter ()
 Modify the output parameter. More...
 
size_t OutputSize () const
 Get the output size. More...
 
OutputDataType const & Parameters () const
 Get the parameters. More...
 
OutputDataType & Parameters ()
 Modify the parameters. More...
 
void Reset ()
 
template<typename Archive >
void serialize (Archive &ar, const unsigned int)
 Serialize the layer. More...
 
OutputDataType const & Weight () const
 Get the weight of the layer. More...
 
OutputDataType & Weight ()
 Modify the weight of the layer. More...
 

Detailed Description

template<typename InputDataType, typename OutputDataType, typename RegularizerType>
class mlpack::ann::Linear3D< InputDataType, OutputDataType, RegularizerType >

Implementation of the Linear3D layer class.

The Linear class represents a single layer of a neural network.

Shape of input : (inSize * nPoints, batchSize) Shape of output : (outSize * nPoints, batchSize)

Template Parameters
InputDataTypeType of the input data (arma::colvec, arma::mat, arma::sp_mat or arma::cube).
OutputDataTypeType of the output data (arma::colvec, arma::mat, arma::sp_mat or arma::cube).

Definition at line 105 of file layer_types.hpp.

Constructor & Destructor Documentation

Linear3D ( )

Create the Linear3D object.

Linear3D ( const size_t  inSize,
const size_t  outSize,
RegularizerType  regularizer = RegularizerType() 
)

Create the Linear3D layer object using the specified number of units.

Parameters
inSizeThe number of input units.
outSizeThe number of output units.
regularizerThe regularizer to use, optional.

Member Function Documentation

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 backwards trough f.

Using the results from the feed forward pass.

Parameters
*(input) The propagated input activation.
gyThe backpropagated error.
gThe calculated gradient.
OutputDataType const& Bias ( ) const
inline

Get the bias of the layer.

Definition at line 135 of file linear3d.hpp.

OutputDataType& Bias ( )
inline

Modify the bias weights of the layer.

Definition at line 137 of file linear3d.hpp.

OutputDataType const& Delta ( ) const
inline

Get the delta.

Definition at line 114 of file linear3d.hpp.

OutputDataType& Delta ( )
inline

Modify the delta.

Definition at line 116 of file linear3d.hpp.

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 activity forward through f.

Parameters
inputInput data used for evaluating the specified function.
outputResulting output activation.
void Gradient ( const arma::Mat< eT > &  input,
const arma::Mat< eT > &  error,
arma::Mat< eT > &  gradient 
)
OutputDataType const& Gradient ( ) const
inline

Get the gradient.

Definition at line 125 of file linear3d.hpp.

OutputDataType& Gradient ( )
inline

Modify the gradient.

Definition at line 127 of file linear3d.hpp.

InputDataType const& InputParameter ( ) const
inline

Get the input parameter.

Definition at line 104 of file linear3d.hpp.

InputDataType& InputParameter ( )
inline

Modify the input parameter.

Definition at line 106 of file linear3d.hpp.

size_t InputSize ( ) const
inline

Get the input size.

Definition at line 119 of file linear3d.hpp.

OutputDataType const& OutputParameter ( ) const
inline

Get the output parameter.

Definition at line 109 of file linear3d.hpp.

OutputDataType& OutputParameter ( )
inline

Modify the output parameter.

Definition at line 111 of file linear3d.hpp.

size_t OutputSize ( ) const
inline

Get the output size.

Definition at line 122 of file linear3d.hpp.

OutputDataType const& Parameters ( ) const
inline

Get the parameters.

Definition at line 99 of file linear3d.hpp.

OutputDataType& Parameters ( )
inline

Modify the parameters.

Definition at line 101 of file linear3d.hpp.

void Reset ( )
void serialize ( Archive &  ar,
const unsigned  int 
)

Serialize the layer.

OutputDataType const& Weight ( ) const
inline

Get the weight of the layer.

Definition at line 130 of file linear3d.hpp.

OutputDataType& Weight ( )
inline

Modify the weight of the layer.

Definition at line 132 of file linear3d.hpp.


The documentation for this class was generated from the following files: