MagmaDNN  1.0
c++NeuralNetworkFramework
dropoutlayer.h File Reference
#include <vector>
#include "layer/layer.h"
#include "tensor/tensor.h"
#include "compute/operation.h"
#include "compute/tensor_operations.h"
Include dependency graph for dropoutlayer.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  magmadnn::layer::DropoutLayer< T >
 

Functions

template<typename T >
DropoutLayer< T > * magmadnn::layer::dropout (op::Operation< T > *input, float dropout_rate, unsigned long long seed)
 

Variables

const float DEFAULT_DROPOUT_RATE = 0.2
 
const unsigned long long DEFAULT_SEED = time(NULL)
 

Detailed Description

Author
Sedrick Keh
Version
1.0
Date
2019-06-28

Function Documentation

◆ dropout()

template<typename T >
DropoutLayer< T > * magmadnn::layer::dropout ( op::Operation< T > *  input,
float  dropout_rate = DEFAULT_DROPOUT_RATE,
unsigned long long  seed = DEFAULT_SEED 
)

A new Dropout layer.

Template Parameters
Tnumeric
Parameters
inputinput tensor to randomly dropout.
dropout_ratepercentage of values to dropout.
seedseed for random sampling.
Returns
DropoutLayer<T>* a dropout layer.