MagmaDNN  1.0
c++NeuralNetworkFramework
sigmoid_internal.h File Reference
#include <math.h>
#include "tensor/tensor.h"
Include dependency graph for sigmoid_internal.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::internal::sigmoid_full (Tensor< T > *x, Tensor< T > *out, bool fast)
 
template<typename T >
void magmadnn::internal::sigmoid_grad (Tensor< T > *output, Tensor< T > *grad, Tensor< T > *out)
 

Detailed Description

Author
Daniel Nichols
Version
1.0
Date
2019-02-23

Function Documentation

◆ sigmoid_full()

template<typename T >
void magmadnn::internal::sigmoid_full ( Tensor< T > *  x,
Tensor< T > *  out,
bool  fast = true 
)

Computes the element-wise sigmoid on x.

Template Parameters
T
Parameters
xpointer to tensor to be sigmoided
fastif true, then x=1/(1+|x|) is computed instead of normal sigmoid