MagmaDNN  1.0
c++NeuralNetworkFramework
dropout.h File Reference
#include "tensor/tensor.h"
#include "math/product.h"
#include "utilities_internal.h"
#include "types.h"
Include dependency graph for dropout.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions

template<typename T >
void magmadnn::math::dropout (Tensor< T > *x, Tensor< T > *out, Tensor< T > *mask, float dropout_rate)
 
template<typename T >
void magmadnn::math::dropout_grad (Tensor< T > *grad, Tensor< T > *out, Tensor< T > *mask)
 

Detailed Description

Author
Sedrick Keh
Version
0.1
Date
2019-06-28