iganet::SELU Class Reference

SELU activation function. More...

#include </home/runner/work/iganet/iganet/include/iganet/net/activation.hpp>

Inheritance diagram for iganet::SELU:

Public Member Functions
	SELU (bool inplace)
	Provides the SELU operation.
	SELU (torch::nn::functional::SELUFuncOptions options={})
	Provides the SELU operation.
	~SELU () override=default
torch::Tensor	apply (const torch::Tensor &input) const override
	Applies the activation function to the given input.
torch::nn::functional::SELUFuncOptions &	options ()
	Returns non-constant reference to options.
const torch::nn::functional::SELUFuncOptions &	options () const
	Returns constant reference to options.
void	pretty_print (std::ostream &os) const noexcept override
	Returns a string representation of the activation function.
torch::serialize::InputArchive &	read (torch::serialize::InputArchive &archive, const std::string &key="selu") override
	Reads the activation function from a torch::serialize::InputArchive object.
torch::serialize::OutputArchive &	write (torch::serialize::OutputArchive &archive, const std::string &key="selu") const override
	Writes the activation function into a torch::serialize::OutputArchive object.
Public Member Functions inherited from iganet::ActivationFunction
	~ActivationFunction () override=default

Private Attributes
torch::nn::functional::SELUFuncOptions	options_

Additional Inherited Members
Protected Member Functions inherited from iganet::utils::FullQualifiedName
virtual	~FullQualifiedName ()=default
virtual const std::string &	name () const noexcept
	Returns the full qualified name of the object.
Protected Attributes inherited from iganet::utils::FullQualifiedName
at::optional< std::string >	name_
	String storing the full qualified name of the object.

Detailed Description

SELU activation function.

\[ \text{SELU}(x) = s * ( \max(0,x) + \min(0, \alpha*(\exp(x)-1 ) ) ) \]

with \( s = 1.0507009873554804934193349852946 \) and. \( \alpha = 1.6732632423543772848170429916717 \).

Constructor & Destructor Documentation

SELU() [1/2]

iganet::SELU::SELU ( torch::nn::functional::SELUFuncOptions  options = {} )

inlineexplicit

Provides the SELU operation.

Parameters

options

Configuration options.

SELU() [2/2]

iganet::SELU::SELU ( bool  inplace )

inlineexplicit

Provides the SELU operation.

Parameters

inplace

Value of inplace.

~SELU()

iganet::SELU::~SELU ( )

overridedefault

Member Function Documentation

apply()

torch::Tensor iganet::SELU::apply ( const torch::Tensor &  input ) const

inlineoverridevirtual

Applies the activation function to the given input.

Parameters

input

Tensor to which the activation function is applied.

Returns

The activated tensor.

Implements iganet::ActivationFunction.

options() [1/2]

torch::nn::functional::SELUFuncOptions & iganet::SELU::options ( )

inline

Returns non-constant reference to options.

Returns

Mutable reference to the activation options.

options() [2/2]

const torch::nn::functional::SELUFuncOptions & iganet::SELU::options ( ) const

inline

Returns constant reference to options.

Returns

Constant reference to the activation options.

pretty_print()

void iganet::SELU::pretty_print ( std::ostream &  os ) const

inlineoverridevirtualnoexcept

Returns a string representation of the activation function.

Parameters

os	Stream that receives the representation.

Implements iganet::ActivationFunction.

read()

torch::serialize::InputArchive & iganet::SELU::read ( torch::serialize::InputArchive &  archive, const std::string &  key = "selu"  )

inlineoverridevirtual

Reads the activation function from a torch::serialize::InputArchive object.

Parameters

archive	Input archive from which the activation state is read.
key	Key prefix under which the state is stored.

Returns

A reference to archive.

Implements iganet::ActivationFunction.

write()

torch::serialize::OutputArchive & iganet::SELU::write ( torch::serialize::OutputArchive &  archive, const std::string &  key = "selu"  ) const

inlineoverridevirtual

Writes the activation function into a torch::serialize::OutputArchive object.

Parameters

archive	Output archive that receives the activation state.
key	Key prefix under which the state is stored.

Returns

A reference to archive.

Implements iganet::ActivationFunction.

Member Data Documentation

options_

torch::nn::functional::SELUFuncOptions iganet::SELU::options_

private

---

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

• include/iganet/net/activation.hpp