Randomized ReLU activation function.
More...
#include </home/runner/work/iganet/iganet/include/layer.hpp>
|
| RReLU (double lower, double upper, bool inplace=false) |
|
| RReLU (torch::nn::functional::RReLUFuncOptions options={}) |
|
| ~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.
|
|
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="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.
|
|
virtual | ~ActivationFunction ()=default |
|
|
torch::nn::functional::RReLUFuncOptions | 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.
|
|
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 |
( |
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()
virtual void iganet::RReLU::pretty_print |
( |
std::ostream & |
os = Log(log::info) | ) |
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: