cupyx.scipy.special.log_softmax#
- cupyx.scipy.special.log_softmax(x, axis=None)[source]#
Compute logarithm of softmax function
- Parameters
x (array-like) – Input array
axis (int or tuple of ints, optional) – Axis to compute values along. Default is None and softmax will be computed over the entire array x
- Returns
s – An array with the same shape as x. Exponential of the result will sum to 1 along the specified axis. If x is a scalar, a scalar is returned
- Return type
cupy.ndarry