#include <math.h>
#include "tensor/tensor.h"
Go to the source code of this file.
|
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) |
|
- Author
- Daniel Nichols
- Version
- 1.0
- Date
- 2019-02-23
- Copyright
- Copyright (c) 2019
◆ 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
-
- Parameters
-
x | pointer to tensor to be sigmoided |
fast | if true, then x=1/(1+|x|) is computed instead of normal sigmoid |