#include <vector>
#include "layer/layer.h"
#include "tensor/tensor.h"
#include "compute/operation.h"
#include "compute/tensor_operations.h"
Go to the source code of this file.
|
template<typename T > |
DropoutLayer< T > * | magmadnn::layer::dropout (op::Operation< T > *input, float dropout_rate, unsigned long long seed) |
|
|
const float | DEFAULT_DROPOUT_RATE = 0.2 |
|
const unsigned long long | DEFAULT_SEED = time(NULL) |
|
- Author
- Sedrick Keh
- Version
- 1.0
- Date
- 2019-06-28
- Copyright
- Copyright (c) 2019
◆ 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
-
- Parameters
-
input | input tensor to randomly dropout. |
dropout_rate | percentage of values to dropout. |
seed | seed for random sampling. |
- Returns
- DropoutLayer<T>* a dropout layer.