base_layer.hpp File Reference
Include dependency graph for base_layer.hpp:
This graph shows which files directly or indirectly include this file:
Go to the source code of this file.
Classes | |
| class | BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Implementation of the base layer. More... | |
Namespaces | |
| mlpack | |
strip_type.hpp | |
| mlpack::ann | |
Artificial Neural Network. | |
Typedefs | |
template < class ActivationFunction = GELUFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | GELUFunctionLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard GELU-Layer using the GELU activation function. More... | |
template < class ActivationFunction = HardSigmoidFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | HardSigmoidLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard HardSigmoid-Layer using the HardSigmoid activation function. More... | |
template < class ActivationFunction = IdentityFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | IdentityLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard Identity-Layer using the identity activation function. More... | |
template < class ActivationFunction = LiSHTFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | LiSHTFunctionLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard LiSHT-Layer using the LiSHT activation function. More... | |
template < class ActivationFunction = MishFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | MishFunctionLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard Mish-Layer using the Mish activation function. More... | |
template < class ActivationFunction = RectifierFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | ReLULayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard rectified linear unit non-linearity layer. More... | |
template < class ActivationFunction = LogisticFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | SigmoidLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard Sigmoid-Layer using the logistic activation function. More... | |
template < class ActivationFunction = SoftplusFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | SoftPlusLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard Softplus-Layer using the Softplus activation function. More... | |
template < class ActivationFunction = SwishFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | SwishFunctionLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard Swish-Layer using the Swish activation function. More... | |
template < class ActivationFunction = TanhFunction , typename InputDataType = arma::mat , typename OutputDataType = arma::mat > | |
| using | TanHLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType > |
| Standard hyperbolic tangent layer. More... | |
Detailed Description
Definition of the BaseLayer class, which attaches various functions to the embedding layer.
mlpack is free software; you may redistribute it and/or modify it under the terms of the 3-clause BSD license. You should have received a copy of the 3-clause BSD license along with mlpack. If not, see http://www.opensource.org/licenses/BSD-3-Clause for more information.
Definition in file base_layer.hpp.