MLask 1.0.0
A custom c++ deep learning library
Loading...
Searching...
No Matches
Public Member Functions | List of all members
mlask::Sigmoid Class Reference

class reprsenting Sigmoid activation function More...

#include <Sigmoid.hpp>

Inheritance diagram for mlask::Sigmoid:
Inheritance graph
[legend]
Collaboration diagram for mlask::Sigmoid:
Collaboration graph
[legend]

Public Member Functions

 Sigmoid (std::size_t in=0)
 Construct a new Sigmoid 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 reprsenting Sigmoid activation function

Constructor & Destructor Documentation

◆ Sigmoid()

mlask::Sigmoid::Sigmoid ( std::size_t  in = 0)

Construct a new Sigmoid object.

Parameters
inNumber of inputs to the layer, default is 0, meaning it can be any size

Member Function Documentation

◆ str()

std::string mlask::Sigmoid::str ( ) const
inlineoverridevirtual

Returns a string representation of the layer.

Reimplemented from mlask::ActivationFunction.

◆ tryConvertToONNX()

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

Try to convert the layer to ONNX format.

Parameters
graphONNX graph to which the layer should be added
inputName of the input tensor
outputName 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: