Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
opennn::DropoutOperator Struct Reference

#include <opennn/network/operators/dropout_operator.h>

Inheritance diagram for opennn::DropoutOperator:
opennn::Operator

Public Member Functions

bool active () const
 
void set_rate (float)
 
void forward_propagate (ForwardPropagation &, size_t, ForwardPropagationMode) override
 
void back_propagate (ForwardPropagation &, BackPropagation &, size_t) const override
 
void to_JSON (JsonWriter &) const override
 
void from_JSON (const Json *) override
 
- Public Member Functions inherited from opennn::Operator
virtual ~Operator ()=default
 
virtual vector< TensorSpec > parameter_specs () const
 
virtual vector< TensorSpec > state_specs () const
 
virtual vector< SlotQuantization > parameter_quantization () const
 
virtual vector< ParameterSlot > parameter_slots ()
 
virtual void link_parameters (span< const TensorView > views)
 
virtual void link_gradients (span< const TensorView > views)
 
virtual void link_states (span< const TensorView >)
 
virtual void link_parameter_scales (span< const TensorView >)
 
bool link_slots (span< const TensorView > views, TensorView *ParameterSlot::*member)
 
virtual void initialize_states ()
 
virtual void set_weights_dtype (Type new_weights_dtype)
 
virtual void set_parameters_random ()
 
virtual void set_parameters_glorot ()
 
virtual void set_parameters_pytorch ()
 
virtual void load_state_from_JSON (const Json *)
 
TensorView & get_input (ForwardPropagation &forward_propagation, size_t layer, size_t slot_index=0) const noexcept
 
vector< TensorView > & get_inputs (ForwardPropagation &forward_propagation, size_t layer) const noexcept
 
TensorView & get_output (ForwardPropagation &forward_propagation, size_t layer, size_t slot_index=0) const noexcept
 
TensorView & get_output_delta (BackPropagation &back_propagation, size_t layer, size_t slot_index=0) const noexcept
 
TensorView & get_input_delta (BackPropagation &back_propagation, size_t layer, size_t slot_index=0) const noexcept
 

Public Attributes

float rate = 0.0f
 
optional< size_t > mask_slot
 
- Public Attributes inherited from opennn::Operator
Type compute_dtype = Type::FP32
 
Type weights_dtype = Type::FP32
 
vector< size_t > input_slots = {0}
 
vector< size_t > output_slots = {1}
 
vector< size_t > input_delta_slots = {1}
 
vector< size_t > output_delta_slots = {0}
 

Member Function Documentation

◆ active()

bool opennn::DropoutOperator::active ( ) const
inline

◆ back_propagate()

void opennn::DropoutOperator::back_propagate ( ForwardPropagation & ,
BackPropagation & ,
size_t  ) const
overridevirtual

Reimplemented from opennn::Operator.

◆ forward_propagate()

void opennn::DropoutOperator::forward_propagate ( ForwardPropagation & ,
size_t ,
ForwardPropagationMode  )
overridevirtual

Reimplemented from opennn::Operator.

◆ from_JSON()

void opennn::DropoutOperator::from_JSON ( const Json * )
overridevirtual

Reimplemented from opennn::Operator.

◆ set_rate()

void opennn::DropoutOperator::set_rate ( float )

◆ to_JSON()

void opennn::DropoutOperator::to_JSON ( JsonWriter & ) const
overridevirtual

Reimplemented from opennn::Operator.

Member Data Documentation

◆ mask_slot

optional<size_t> opennn::DropoutOperator::mask_slot

◆ rate

float opennn::DropoutOperator::rate = 0.0f