Method Softmax
Softmax(TensorFloat, TensorFloat, int)
Computes an output tensor by applying the Softmax activation function along an axis: f(x, axis) = exp(X) / ReduceSum(exp(X), axis).
Declaration
void Softmax(TensorFloat X, TensorFloat O, int axis)Parameters
| Type | Name | Description |
|---|---|---|
| TensorFloat | X | The input tensor. |
| TensorFloat | O | The output tensor to be computed and filled. |
| int | axis | The axis along which to apply the |