class ngraph::op::v0::Selu

Overview

Performs a SELU activation function on all elements of the input node. More…

#include <selu.hpp>

class Selu: public ngraph::op::util::FusedOp
{
public:
    // fields

    static constexpr NodeTypeInfo type_info {"Selu", 0};

    // construction

    Selu();

    Selu(
        const Output<Node>& data,
        const Output<Node>& alpha,
        const Output<Node>& lambda
        );

    // methods

    const NodeTypeInfo& get_type_info() const;
    virtual void pre_validate_and_infer_types();
    bool visit_attributes(AttributeVisitor& visitor);
    virtual OutputVector decompose_op() const;
    virtual std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const;
};

Detailed Documentation

Performs a SELU activation function on all elements of the input node.

Construction

Selu(
    const Output<Node>& data,
    const Output<Node>& alpha,
    const Output<Node>& lambda
    )

Constructs a Selu node.

Parameters:

data

  • Node producing the input tensor

alpha

  • Alpha coefficient of SELU operation

lambda

  • Lambda coefficient of SELU operation