mlask::LeakyRelu Class Reference

Class representing Leaky Relu activation function. More...

#include <LeakyRelu.hpp>

Inheritance diagram for mlask::LeakyRelu:

Collaboration diagram for mlask::LeakyRelu:

Public Member Functions
	LeakyRelu (float_t a, std::size_t in=0)
	Construct a new LeakyRelu object.
bool	tryConvertToONNX (onnx::GraphProto *graph, std::string input, std::string output) const override
	Try to convert the layer to ONNX format.
std::string	str () const override
	Returns a string representation of the layer.

Detailed Description

Class representing Leaky Relu activation function.

Constructor & Destructor Documentation

LeakyRelu()

mlask::LeakyRelu::LeakyRelu	(	float_t	a,
		std::size_t	in = 0
	)

Construct a new LeakyRelu object.

Parameters

a	an a for leaky relu definition
in	Number of inputs to the layer, default is 0, meaning it can be any size

Member Function Documentation

str()

std::string mlask::LeakyRelu::str ( ) const

inlineoverridevirtual

Returns a string representation of the layer.

Reimplemented from mlask::ActivationFunction.

tryConvertToONNX()

bool mlask::LeakyRelu::tryConvertToONNX ( onnx::GraphProto *  graph, std::string  input, std::string  output  ) const

overridevirtual

Try to convert the layer to ONNX format.

Parameters

graph	ONNX graph to which the layer should be added
input	Name of the input tensor
output	Name of the output tensor

Returns

True if the conversion was successful, false otherwise

Reimplemented from mlask::Layer.

---

The documentation for this class was generated from the following files:

• include/activation_functions/LeakyRelu.hpp
• src/activation_functions/LeakyRelu.cpp