class ngraph::op::v0::HardSigmoid

Overview

Parameterized, bounded sigmoid-like, piecewise linear function. min(max(alpha*x + beta, 0), 1) More…

#include <hard_sigmoid.hpp>

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

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

    // construction

    HardSigmoid();

    HardSigmoid(
        const Output<Node>& data,
        const Output<Node>& alpha,
        const Output<Node>& beta
        );

    // methods

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

Detailed Documentation

Parameterized, bounded sigmoid-like, piecewise linear function. min(max(alpha*x + beta, 0), 1)

Construction

HardSigmoid(
    const Output<Node>& data,
    const Output<Node>& alpha,
    const Output<Node>& beta
    )

Constructs a HardSigmoid operation.

Parameters:

data

Input tensor.

alpha

A scalar value representing the alpha parameter.

beta

A scalar value representing the beta parameter.