MLask/Relu_8hpp_source.html

#pragma once

#include "ActivationFunction.hpp"

#include "Layer.hpp"

#include <Eigen/Core>


namespace mlask{


class Relu : public ActivationFunction{

    vectorIn input_;

    float_t activate(float_t input)override;

    float_t derived(float_t input)override;

public:

    Relu(std::size_t in = 0);


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

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

};


}

ActivationFunction.hpp

Layer.hpp

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

mlask::Relu
Layer representing Relu activation function.
Definition Relu.hpp:8

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

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

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