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

Lp Normalization. More...

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

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

Public Member Functions

 Normalize (double p, double eps, int64_t dim)
 
 Normalize (torch::nn::functional::NormalizeFuncOptions options={})
 
 ~Normalize () override=default
 
torch::Tensor apply (const torch::Tensor &input) const override
 Applies the activation function to the given input.
 
torch::nn::functional::NormalizeFuncOptions & options ()
 Returns non-constant reference to options.
 
const torch::nn::functional::NormalizeFuncOptions & 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="normalize") override
 Reads the activation function from a torch::serialize::InputArchive object.
 
torch::serialize::OutputArchive & write (torch::serialize::OutputArchive &archive, const std::string &key="normalize") 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::NormalizeFuncOptions 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

Lp Normalization.

Constructor & Destructor Documentation

◆ Normalize() [1/2]

iganet::Normalize::Normalize ( torch::nn::functional::NormalizeFuncOptions  options = {})
inlineexplicit

◆ Normalize() [2/2]

iganet::Normalize::Normalize ( double  p,
double  eps,
int64_t  dim 
)
inlineexplicit

◆ ~Normalize()

iganet::Normalize::~Normalize ( )
overridedefault

Member Function Documentation

◆ apply()

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

Applies the activation function to the given input.

Implements iganet::ActivationFunction.

◆ options() [1/2]

torch::nn::functional::NormalizeFuncOptions & iganet::Normalize::options ( )
inline

Returns non-constant reference to options.

◆ options() [2/2]

const torch::nn::functional::NormalizeFuncOptions & iganet::Normalize::options ( ) const
inline

Returns constant reference to options.

◆ pretty_print()

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

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

Implements iganet::ActivationFunction.

◆ write()

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

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

Implements iganet::ActivationFunction.

Member Data Documentation

◆ options_

torch::nn::functional::NormalizeFuncOptions iganet::Normalize::options_
private

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