iganet::SiLU Class Reference

Sigmoid Linear Unit activation function. More...

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

Inheritance diagram for iganet::SiLU:

Public Member Functions
torch::Tensor	apply (const torch::Tensor &input) const override
	Applies the activation function to the given input.
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="silu") override
	Reads the activation function from a torch::serialize::InputArchive object.
torch::serialize::OutputArchive &	write (torch::serialize::OutputArchive &archive, const std::string &key="silu") const override
	Writes the activation function into a torch::serialize::OutputArchive object.
Public Member Functions inherited from iganet::ActivationFunction
	~ActivationFunction () override=default

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

Sigmoid Linear Unit activation function.

\[ \text{SiLU}(x) = x * \sigma(x) \]

Member Function Documentation

apply()

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

inlineoverridevirtual

Applies the activation function to the given input.

Implements iganet::ActivationFunction.

pretty_print()

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

inlineoverridevirtualnoexcept

Returns a string representation of the activation function.

Implements iganet::ActivationFunction.

read()

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

inlineoverridevirtual

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

Implements iganet::ActivationFunction.

write()

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

inlineoverridevirtual

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

Implements iganet::ActivationFunction.

---

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

• include/net/activation.hpp