IgANet
IgANets - Isogeometric Analysis Networks
Loading...
Searching...
No Matches
iganet::RReLU Class Reference

Randomized ReLU activation function. More...

#include </home/runner/work/iganet/iganet/include/layer.hpp>

Inheritance diagram for iganet::RReLU:
iganet::ActivationFunction iganet::utils::FullQualifiedName

Public Member Functions

 RReLU (double lower, double upper, bool inplace=false)
 
 RReLU (torch::nn::functional::RReLUFuncOptions options={})
 
 ~RReLU () override=default
 
torch::Tensor apply (const torch::Tensor &input) const override
 Applies the activation function to the given input.
 
torch::nn::functional::RReLUFuncOptions & options ()
 Returns non-constant reference to options.
 
const torch::nn::functional::RReLUFuncOptions & options () const
 Returns constant reference to options.
 
virtual void pretty_print (std::ostream &os=Log(log::info)) const noexcept override
 Returns a string representation of the activation function.
 
torch::serialize::InputArchive & read (torch::serialize::InputArchive &archive, const std::string &key="rrelu") override
 Reads the activation function from a torch::serialize::InputArchive object.
 
torch::serialize::OutputArchive & write (torch::serialize::OutputArchive &archive, const std::string &key="rrelu") const override
 Writes the activation function into a torch::serialize::OutputArchive object.
 
- Public Member Functions inherited from iganet::ActivationFunction
virtual ~ActivationFunction ()=default
 

Private Attributes

torch::nn::functional::RReLUFuncOptions options_
 

Additional Inherited Members

- Protected Member Functions inherited from iganet::utils::FullQualifiedName
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

Randomized ReLU activation function.

\[ \text{RReLU}(x) = \begin{cases} x & \text{ if } x \ge 0 a * x & \text{ otherwise } \end{cases} \]

Constructor & Destructor Documentation

◆ RReLU() [1/2]

iganet::RReLU::RReLU ( torch::nn::functional::RReLUFuncOptions  options = {})
inlineexplicit

◆ RReLU() [2/2]

iganet::RReLU::RReLU ( double  lower,
double  upper,
bool  inplace = false 
)
inlineexplicit

◆ ~RReLU()

iganet::RReLU::~RReLU ( )
overridedefault

Member Function Documentation

◆ apply()

torch::Tensor iganet::RReLU::apply ( const torch::Tensor &  input) const
inlineoverridevirtual

Applies the activation function to the given input.

Implements iganet::ActivationFunction.

◆ options() [1/2]

torch::nn::functional::RReLUFuncOptions & iganet::RReLU::options ( )
inline

Returns non-constant reference to options.

◆ options() [2/2]

const torch::nn::functional::RReLUFuncOptions & iganet::RReLU::options ( ) const
inline

Returns constant reference to options.

◆ pretty_print()

virtual void iganet::RReLU::pretty_print ( std::ostream &  os = Log(log::info)) const
inlineoverridevirtualnoexcept

Returns a string representation of the activation function.

Implements iganet::ActivationFunction.

◆ read()

torch::serialize::InputArchive & iganet::RReLU::read ( torch::serialize::InputArchive &  archive,
const std::string &  key = "rrelu" 
)
inlineoverridevirtual

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

Implements iganet::ActivationFunction.

◆ write()

torch::serialize::OutputArchive & iganet::RReLU::write ( torch::serialize::OutputArchive &  archive,
const std::string &  key = "rrelu" 
) const
inlineoverridevirtual

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

Implements iganet::ActivationFunction.

Member Data Documentation

◆ options_

torch::nn::functional::RReLUFuncOptions iganet::RReLU::options_
private

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