Softmin activation function.
More...
#include </home/runner/work/iganet/iganet/include/iganet/net/activation.hpp>
|
| | Softmin (const torch::nn::functional::SoftminFuncOptions &options) |
| | Provides the Softmin operation.
|
| |
| | Softmin (int64_t dim) |
| | Provides the Softmin operation.
|
| |
| | ~Softmin () override=default |
| |
| torch::Tensor | apply (const torch::Tensor &input) const override |
| | Applies the activation function to the given input.
|
| |
| torch::nn::functional::SoftminFuncOptions & | options () |
| | Returns non-constant reference to options.
|
| |
| const torch::nn::functional::SoftminFuncOptions & | 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="softmin") override |
| | Reads the activation function from a torch::serialize::InputArchive object.
|
| |
| torch::serialize::OutputArchive & | write (torch::serialize::OutputArchive &archive, const std::string &key="softmin") const override |
| | Writes the activation function into a torch::serialize::OutputArchive object.
|
| |
| | ~ActivationFunction () override=default |
| |
|
| torch::nn::functional::SoftminFuncOptions | 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.
|
| |
Softmin activation function.
\[
\text{Softmin}(x) = \text{Softmax}(-x)
\]
◆ Softmin() [1/2]
| iganet::Softmin::Softmin |
( |
int64_t |
dim | ) |
|
|
inlineexplicit |
Provides the Softmin operation.
- Parameters
-
◆ Softmin() [2/2]
| iganet::Softmin::Softmin |
( |
const torch::nn::functional::SoftminFuncOptions & |
options | ) |
|
|
inlineexplicit |
Provides the Softmin operation.
- Parameters
-
| options | Configuration options. |
◆ ~Softmin()
| iganet::Softmin::~Softmin |
( |
| ) |
|
|
overridedefault |
◆ apply()
| torch::Tensor iganet::Softmin::apply |
( |
const torch::Tensor & |
input | ) |
const |
|
inlineoverridevirtual |
Applies the activation function to the given input.
- Parameters
-
| input | Tensor to which the activation function is applied. |
- Returns
- The activated tensor.
Implements iganet::ActivationFunction.
◆ options() [1/2]
| torch::nn::functional::SoftminFuncOptions & iganet::Softmin::options |
( |
| ) |
|
|
inline |
Returns non-constant reference to options.
- Returns
- Mutable reference to the activation options.
◆ options() [2/2]
| const torch::nn::functional::SoftminFuncOptions & iganet::Softmin::options |
( |
| ) |
const |
|
inline |
Returns constant reference to options.
- Returns
- Constant reference to the activation options.
◆ pretty_print()
| void iganet::Softmin::pretty_print |
( |
std::ostream & |
os | ) |
const |
|
inlineoverridevirtualnoexcept |
Returns a string representation of the activation function.
- Parameters
-
| os | Stream that receives the representation. |
Implements iganet::ActivationFunction.
◆ read()
| torch::serialize::InputArchive & iganet::Softmin::read |
( |
torch::serialize::InputArchive & |
archive, |
|
|
const std::string & |
key = "softmin" |
|
) |
| |
|
inlineoverridevirtual |
Reads the activation function from a torch::serialize::InputArchive object.
- Parameters
-
| archive | Input archive from which the activation state is read. |
| key | Key prefix under which the state is stored. |
- Returns
- A reference to
archive.
Implements iganet::ActivationFunction.
◆ write()
| torch::serialize::OutputArchive & iganet::Softmin::write |
( |
torch::serialize::OutputArchive & |
archive, |
|
|
const std::string & |
key = "softmin" |
|
) |
| const |
|
inlineoverridevirtual |
Writes the activation function into a torch::serialize::OutputArchive object.
- Parameters
-
| archive | Output archive that receives the activation state. |
| key | Key prefix under which the state is stored. |
- Returns
- A reference to
archive.
Implements iganet::ActivationFunction.
◆ options_
| torch::nn::functional::SoftminFuncOptions iganet::Softmin::options_ |
|
private |
The documentation for this class was generated from the following file: