Continuously Differentiable Exponential Linear Units activation function.
More...
#include </home/runner/work/iganet/iganet/include/layer.hpp>
|
| CELU (double alpha, bool inplace=false) |
|
| CELU (torch::nn::functional::CELUFuncOptions options={}) |
|
| ~CELU () override=default |
|
torch::Tensor | apply (const torch::Tensor &input) const override |
| Applies the activation function to the given input.
|
|
torch::nn::functional::CELUFuncOptions & | options () |
| Returns non-constant reference to options.
|
|
const torch::nn::functional::CELUFuncOptions & | 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="celu") override |
| Reads the activation function from a torch::serialize::InputArchive object.
|
|
torch::serialize::OutputArchive & | write (torch::serialize::OutputArchive &archive, const std::string &key="celu") const override |
| Writes the activation function into a torch::serialize::OutputArchive object.
|
|
virtual | ~ActivationFunction ()=default |
|
|
torch::nn::functional::CELUFuncOptions | options_ |
|
|
virtual const std::string & | name () const noexcept |
| Returns the full qualified name of the object.
|
|
at::optional< std::string > | name_ |
| String storing the full qualified name of the object.
|
|
Continuously Differentiable Exponential Linear Units activation function.
\[
\text{CELU}(x) = \max(0,x) + \min(0, \alpha * (\exp(x/\alpha)-1))
\]
◆ CELU() [1/2]
iganet::CELU::CELU |
( |
torch::nn::functional::CELUFuncOptions |
options = {} | ) |
|
|
inlineexplicit |
◆ CELU() [2/2]
iganet::CELU::CELU |
( |
double |
alpha, |
|
|
bool |
inplace = false |
|
) |
| |
|
inlineexplicit |
◆ ~CELU()
◆ apply()
torch::Tensor iganet::CELU::apply |
( |
const torch::Tensor & |
input | ) |
const |
|
inlineoverridevirtual |
◆ options() [1/2]
torch::nn::functional::CELUFuncOptions & iganet::CELU::options |
( |
| ) |
|
|
inline |
Returns non-constant reference to options.
◆ options() [2/2]
const torch::nn::functional::CELUFuncOptions & iganet::CELU::options |
( |
| ) |
const |
|
inline |
Returns constant reference to options.
◆ pretty_print()
virtual void iganet::CELU::pretty_print |
( |
std::ostream & |
os = Log(log::info) | ) |
const |
|
inlineoverridevirtualnoexcept |
◆ read()
torch::serialize::InputArchive & iganet::CELU::read |
( |
torch::serialize::InputArchive & |
archive, |
|
|
const std::string & |
key = "celu" |
|
) |
| |
|
inlineoverridevirtual |
◆ write()
torch::serialize::OutputArchive & iganet::CELU::write |
( |
torch::serialize::OutputArchive & |
archive, |
|
|
const std::string & |
key = "celu" |
|
) |
| const |
|
inlineoverridevirtual |
◆ options_
torch::nn::functional::CELUFuncOptions iganet::CELU::options_ |
|
private |
The documentation for this class was generated from the following file: