#include <string>
#include "compute/operation.h"
#include "compute/add/addop.h"
#include "compute/dot/dotop.h"
#include "compute/variable.h"
#include "compute/negative/negativeop.h"
#include "sigmoid_internal.h"
#include "utilities_internal.h"
Go to the source code of this file.
|
template<typename T > |
SigmoidOp< T > * | magmadnn::op::sigmoid (Operation< T > *x, bool copy, bool fast) |
|
- Author
- Daniel Nichols
- Version
- 1.0
- Date
- 2019-02-23
- Copyright
- Copyright (c) 2019
◆ sigmoid()
template<typename T >
SigmoidOp< T > * magmadnn::op::sigmoid |
( |
Operation< T > * |
x, |
|
|
bool |
copy = true , |
|
|
bool |
fast = false |
|
) |
| |
Compute element-wise sigmoid on tensor x.
- Template Parameters
-
- Parameters
-
x | tensor to be computed with. |
copy | if true, a new tensor is allocated and returned. If false, x is overwritten. |
fast | if true, the x=1/(1+|x|) is computed instead of the normal sigmoid function. |
- Returns
- SigmoidOp<T>*