Continuously Differentiable Exponential Linear Units activation function.
More...
#include </home/runner/work/iganet/iganet/include/net/activation.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.
|
| |
| 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="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.
|
| |
| | ~ActivationFunction () override=default |
| |
|
| torch::nn::functional::CELUFuncOptions | options_ |
| |
|
| virtual | ~FullQualifiedName ()=default |
| |
| 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()
| void iganet::CELU::pretty_print |
( |
std::ostream & |
os | ) |
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: