cupyx.scipy.special.softmax#
- cupyx.scipy.special.softmax(x, axis=None)[source]#
Softmax function.
The softmax function transforms each element of a collection by computing the exponential of each element divided by the sum of the exponentials of all the elements.
- Parameters
x (array-like) – The input array
axis (int or tuple of ints, optional) – Axis to compute values along. Default is None
- Returns
s – Returns an array with same shape as input. The result will sum to 1 along the provided axis
- Return type