Method Softmax
Softmax(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
public TensorFloat Softmax(TensorFloat X, int axis = -1)
Parameters
Type | Name | Description |
---|---|---|
TensorFloat | X | The input tensor. |
int | axis | The axis along which to apply the |
Returns
Type | Description |
---|---|
TensorFloat | The computed output tensor. |