#include <relu6.hpp>
|
|
| ReLU6 () |
| | Create the ReLU6 object.
|
| |
| template<typename InputType , typename OutputType > |
| void | Forward (const InputType &input, OutputType &output) |
| | Ordinary feed forward pass of a neural network, evaluating the function f(x) by propagating the activity forward through f. More...
|
| |
| template<typename DataType > |
| void | Backward (const DataType &input, const DataType &gy, DataType &g) |
| | Ordinary feed backward pass of a neural network, calculating the function f(x) by propagating x backwards through f. More...
|
| |
|
OutputDataType const & | OutputParameter () const |
| | Get the output parameter.
|
| |
|
OutputDataType & | OutputParameter () |
| | Modify the output parameter.
|
| |
|
OutputDataType const & | Delta () const |
| | Get the delta.
|
| |
|
OutputDataType & | Delta () |
| | Modify the delta.
|
| |
|
size_t | WeightSize () const |
| | Get size of weights.
|
| |
|
template<typename Archive > |
| void | serialize (Archive &ar, const uint32_t) |
| | Serialize the layer.
|
| |
template<typename InputDataType = arma::mat, typename OutputDataType = arma::mat>
class mlpack::ann::ReLU6< InputDataType, OutputDataType >
- Template Parameters
-
| InputDataType | Type of the input data (arma::colvec, arma::mat, arma::sp_mat or arma::cube). |
| OutputDataType | Type of the output data (arma::colvec, arma::mat, arma::sp_mat or arma::cube). |
◆ Backward()
template<typename InputDataType , typename OutputDataType >
template<typename DataType >
| void mlpack::ann::ReLU6< InputDataType, OutputDataType >::Backward |
( |
const DataType & |
input, |
|
|
const DataType & |
gy, |
|
|
DataType & |
g |
|
) |
| |
Ordinary feed backward pass of a neural network, calculating the function f(x) by propagating x backwards through f.
Using the results from the feed forward pass.
- Parameters
-
| input | The propagated input activation. |
| gy | The backpropagated error. |
| g | The calculated gradient. |
◆ Forward()
template<typename InputDataType , typename OutputDataType >
template<typename InputType , typename OutputType >
| void mlpack::ann::ReLU6< InputDataType, OutputDataType >::Forward |
( |
const InputType & |
input, |
|
|
OutputType & |
output |
|
) |
| |
Ordinary feed forward pass of a neural network, evaluating the function f(x) by propagating the activity forward through f.
- Parameters
-
| input | Input data used for evaluating the specified function. |
| output | Resulting output activation. |
The documentation for this class was generated from the following files:
