MLask/Sigmoid_8hpp_source.html

#pragma once

#include "ActivationFunction.hpp"


namespace mlask{


class Sigmoid : ActivationFunction{

    vectorIn input_;

    float_t activate(float_t input)override;

    float_t derived(float_t input)override;

public:

    Sigmoid(std::size_t in = 0);


    bool tryConvertToONNX(onnx::GraphProto* graph, std::string input, std::string output) const override;

    std::string str() const override{ return "Sigmoid"; }

};


}

ActivationFunction.hpp

mlask::ActivationFunction
base class for Activation Functions, meant for simplicity in definition.
Definition ActivationFunction.hpp:8

mlask::Sigmoid
class reprsenting Sigmoid activation function
Definition Sigmoid.hpp:7

mlask::Sigmoid::str
std::string str() const override
Returns a string representation of the layer.
Definition Sigmoid.hpp:22

mlask::Sigmoid::tryConvertToONNX
bool tryConvertToONNX(onnx::GraphProto *graph, std::string input, std::string output) const override
Try to convert the layer to ONNX format.
Definition Sigmoid.cpp:18

mlask
Definition LeakyRelu.hpp:4

mlask::float_t
float float_t
Definition types.hpp:12

mlask::vectorIn
Eigen::Matrix< float_t, Eigen::Dynamic, 1 > vectorIn
Definition types.hpp:16