mlpack 3.4.2
Classes | Namespaces | Typedefs
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

namespace  mlpack
 Linear algebra utility functions, generally performed on matrices or vectors.
 
namespace  mlpack::ann
 Artificial Neural Network.
 

Typedefs

template<class ActivationFunction = ElishFunction, typename InputDataType = arma::mat, typename OutputDataType = arma::mat>
using ElishFunctionLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType >
 Standard ELiSH-Layer using the ELiSH activation function. More...
 
template<class ActivationFunction = ElliotFunction, typename InputDataType = arma::mat, typename OutputDataType = arma::mat>
using ElliotFunctionLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType >
 Standard Elliot-Layer using the Elliot activation function. More...
 
template<class ActivationFunction = GaussianFunction, typename InputDataType = arma::mat, typename OutputDataType = arma::mat>
using GaussianFunctionLayer = BaseLayer< ActivationFunction, InputDataType, OutputDataType >
 Standard Gaussian-Layer using the Gaussian activation function. More...
 
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

Author
Marcus Edel

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.