Randomized ReLU activation function.
More...
#include </home/runner/work/iganet/iganet/include/net/activation.hpp>
|
| | RReLU (const torch::nn::functional::RReLUFuncOptions &options={}) |
| |
| | RReLU (double lower, double upper, bool inplace=false) |
| |
| | ~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.
|
| |
| 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="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.
|
| |
| | ~ActivationFunction () override=default |
| |
|
| torch::nn::functional::RReLUFuncOptions | 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.
|
| |
Randomized ReLU activation function.
\[
\text{RReLU}(x) =
\begin{cases}
x & \text{ if } x \ge 0
a * x & \text{ otherwise }
\end{cases}
\]
◆ RReLU() [1/2]
| iganet::RReLU::RReLU |
( |
const 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 |
◆ apply()
| torch::Tensor iganet::RReLU::apply |
( |
const torch::Tensor & |
input | ) |
const |
|
inlineoverridevirtual |
◆ 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()
| void iganet::RReLU::pretty_print |
( |
std::ostream & |
os | ) |
const |
|
inlineoverridevirtualnoexcept |
◆ read()
| torch::serialize::InputArchive & iganet::RReLU::read |
( |
torch::serialize::InputArchive & |
archive, |
|
|
const std::string & |
key = "rrelu" |
|
) |
| |
|
inlineoverridevirtual |
◆ write()
| torch::serialize::OutputArchive & iganet::RReLU::write |
( |
torch::serialize::OutputArchive & |
archive, |
|
|
const std::string & |
key = "rrelu" |
|
) |
| const |
|
inlineoverridevirtual |
◆ options_
| torch::nn::functional::RReLUFuncOptions iganet::RReLU::options_ |
|
private |
The documentation for this class was generated from the following file: