Method Shrink
Shrink(TensorFloat, TensorFloat, float, float)
Computes an output tensor by applying the element-wise Shrink activation function: f(x) = x + bias if x < lambd. f(x) = x - bias if x > lambd. Otherwise f(x) = 0.
Declaration
void Shrink(TensorFloat X, TensorFloat O, float bias, float lambd)Parameters
| Type | Name | Description |
|---|---|---|
| TensorFloat | X | The input tensor. |
| TensorFloat | O | The output tensor to be computed and filled. |
| float | bias | The bias value to use for the |
| float | lambd | The lambda value to use for the |