Public Member Functions | |
DropoutLayer (op::Operation< T > *input, float dropout_rate, unsigned long long seed) | |
virtual std::vector< op::Operation< T > * > | get_weights () |
![]() | |
virtual op::Operation< T > * | out () |
op::Operation< T > * | get_input () |
op::Operation< T > * | get_output () |
std::vector< unsigned int > | get_input_shape () const |
std::vector< unsigned int > | get_output_shape () const |
unsigned int | get_input_shape (unsigned int i) const |
unsigned int | get_output_shape (unsigned int i) const |
void | set_name (std::string name) |
std::string | get_name () const |
Protected Member Functions | |
void | init () |
![]() | |
Layer (std::vector< unsigned int > input_shape, op::Operation< T > *input) | |
Protected Attributes | |
float | dropout_rate |
unsigned long long | seed |
![]() | |
std::vector< unsigned int > | input_shape |
std::vector< unsigned int > | output_shape |
op::Operation< T > * | input |
op::Operation< T > * | output |
std::string | name |