|
OpenNN
Open-source neural networks library
|
#include <operators.h>
Public Types | |
| enum class | Function { Identity , Sigmoid , Tanh , ReLU , Softmax } |
Public Member Functions | |
| void | set_function (Function new_function) |
| void | set_function (const string &name) |
| void | forward_propagate (ForwardPropagation &fp, size_t layer, bool is_training) noexcept override |
| void | apply (TensorView &output) |
| void | apply_delta (const TensorView &outputs, TensorView &delta) const |
| void | to_JSON (JsonWriter &w) const override |
| void | from_JSON (const Json *parent) override |
| void | destroy_cuda () override |
| ~Activation () override | |
| Activation ()=default | |
| Activation (const Activation &)=delete | |
| Activation & | operator= (const Activation &)=delete |
Public Member Functions inherited from opennn::Operator | |
| virtual | ~Operator ()=default |
| virtual vector< pair< Shape, Type > > | parameter_specs () const |
| virtual vector< pair< Shape, Type > > | state_specs () const |
| virtual void | link_parameters (const vector< TensorView > &) |
| virtual void | link_gradients (const vector< TensorView > &) |
| virtual void | link_states (const vector< TensorView > &) |
| virtual void | set_parameters_random () |
| virtual void | set_parameters_glorot () |
| virtual void | load_state_from_JSON (const Json *) |
Static Public Member Functions | |
| static const EnumMap< Function > & | map () |
| static Function | from_string (const string &name) |
| static const string & | to_string (Function function) |
| static cudnnActivationMode_t | to_cudnn_mode (Function function) |
Public Attributes | |
| Function | function = Function::Identity |
| cudnnActivationDescriptor_t | descriptor = nullptr |
Public Attributes inherited from opennn::Operator | |
| vector< size_t > | input_slots |
| vector< size_t > | output_slots |
|
strong |
|
inlineoverride |
|
default |
|
delete |
| void opennn::Activation::apply | ( | TensorView & | output | ) |
| void opennn::Activation::apply_delta | ( | const TensorView & | outputs, |
| TensorView & | delta ) const |
|
overridevirtual |
Reimplemented from opennn::Operator.
|
overridevirtualnoexcept |
Reimplemented from opennn::Operator.
|
overridevirtual |
Reimplemented from opennn::Operator.
|
static |
|
delete |
| void opennn::Activation::set_function | ( | const string & | name | ) |
| void opennn::Activation::set_function | ( | Function | new_function | ) |
|
static |
|
overridevirtual |
Reimplemented from opennn::Operator.
|
static |
| cudnnActivationDescriptor_t opennn::Activation::descriptor = nullptr |
| Function opennn::Activation::function = Function::Identity |