class ngraph::op::util::ActivationFunction

Overview

Class representing activation function used in RNN cells. More…

#include <activation_functions.hpp>

class ActivationFunction
{
public:
    // construction

    ActivationFunction(ActivationFunctionType f, float alpha, float beta);
    ActivationFunction(ActivationFunctionType f, float alpha);
    ActivationFunction(ActivationFunctionType f);
    ActivationFunction();

    // methods

    std::shared_ptr<Node> operator () (const std::shared_ptr<Node>& arg) const;
    void set_alpha(float alpha);
    void set_beta(float beta);
};

Detailed Documentation

Class representing activation function used in RNN cells.

Methods

std::shared_ptr<Node> operator () (const std::shared_ptr<Node>& arg) const

Calls stored activation function with provided node argument.