LeakyRelu.hpp

Go to the documentation of this file.

#pragma once
#include "ActivationFunction.hpp"
 
namespace mlask{
 
class LeakyRelu : ActivationFunction{
    float_t a_;
    vectorIn input_;
    float_t activate(float_t input)override;
    float_t derived(float_t input)override;
public:
    LeakyRelu(float_t a, std::size_t in = 0);
 
    bool tryConvertToONNX(onnx::GraphProto* graph, std::string input, std::string output) const override;
    std::string str() const override{ return "Leaky Relu"; }
};
 
}