MNIST::ReLU Class Reference

ActivationFunction ReLU: Rectified Linear Unit. More...

#include <mnist_mlp.hpp>

Static Public Member Functions
static std::vector< Real >	function (std::vector< Real > input)
static std::vector< Real >	derivative (std::vector< Real > input)

Detailed Description

ActivationFunction ReLU: Rectified Linear Unit.

Definition at line 113 of file mnist_mlp.hpp.

Member Function Documentation

static std::vector<Real> MNIST::ReLU::derivative ( std::vector< Real >  input )

inlinestatic

Definition at line 122 of file mnist_mlp.hpp.

    {
        for (auto &i : input) {
            i = i >= 0 ? 1.0 : 0.0;
        }
        return input;
    }

static std::vector<Real> MNIST::ReLU::function ( std::vector< Real >  input )

inlinestatic

Definition at line 115 of file mnist_mlp.hpp.

    {
        for (auto &i : input) {
            i = std::max(0.0, i);
        }
        return input;
    }

---

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

• source/SNABs/mnist/mnist_mlp.hpp