MagmaDNN  1.0
c++NeuralNetworkFramework
activationlayer.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 activationlayer.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  magmadnn::layer::ActivationLayer< T >
 

Enumerations

enum  activation_t { SIGMOID, TANH, RELU, SOFTMAX }
 

Functions

template<typename T >
ActivationLayer< T > * magmadnn::layer::activation (op::Operation< T > *input, activation_t activation_func)
 

Detailed Description

Author
Daniel Nichols
Version
1.0
Date
2019-02-26