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

Local response Normalization. More...

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

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

Public Member Functions

 LocalResponseNorm (const torch::nn::functional::LocalResponseNormFuncOptions &options)
 
 LocalResponseNorm (int64_t size)
 
 LocalResponseNorm (int64_t size, double alpha, double beta, double k)
 
 ~LocalResponseNorm () override=default
 
torch::Tensor apply (const torch::Tensor &input) const override
 Applies the activation function to the given input.
 
torch::nn::functional::LocalResponseNormFuncOptions & options ()
 Returns non-constant reference to options.
 
const torch::nn::functional::LocalResponseNormFuncOptions & 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="local_response_norm") override
 Reads the activation function from a torch::serialize::InputArchive object.
 
torch::serialize::OutputArchive & write (torch::serialize::OutputArchive &archive, const std::string &key="local_response_norm") 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::LocalResponseNormFuncOptions 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

Local response Normalization.

Constructor & Destructor Documentation

◆ LocalResponseNorm() [1/3]

iganet::LocalResponseNorm::LocalResponseNorm ( int64_t  size)
inlineexplicit

◆ LocalResponseNorm() [2/3]

iganet::LocalResponseNorm::LocalResponseNorm ( const torch::nn::functional::LocalResponseNormFuncOptions &  options)
inlineexplicit

◆ LocalResponseNorm() [3/3]

iganet::LocalResponseNorm::LocalResponseNorm ( int64_t  size,
double  alpha,
double  beta,
double  k 
)
inlineexplicit

◆ ~LocalResponseNorm()

iganet::LocalResponseNorm::~LocalResponseNorm ( )
overridedefault

Member Function Documentation

◆ apply()

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

Applies the activation function to the given input.

Implements iganet::ActivationFunction.

◆ options() [1/2]

torch::nn::functional::LocalResponseNormFuncOptions & iganet::LocalResponseNorm::options ( )
inline

Returns non-constant reference to options.

◆ options() [2/2]

const torch::nn::functional::LocalResponseNormFuncOptions & iganet::LocalResponseNorm::options ( ) const
inline

Returns constant reference to options.

◆ pretty_print()

virtual void iganet::LocalResponseNorm::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::LocalResponseNorm::read ( torch::serialize::InputArchive &  archive,
const std::string &  key = "local_response_norm" 
)
inlineoverridevirtual

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

Implements iganet::ActivationFunction.

◆ write()

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

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

Implements iganet::ActivationFunction.

Member Data Documentation

◆ options_

torch::nn::functional::LocalResponseNormFuncOptions iganet::LocalResponseNorm::options_
private

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