scipy.special.

log_softmax

scipy.special.log_softmax(x, axis=None)

计算 softmax 函数的对数。

原则上是

log_softmax(x) = log(softmax(x))

但使用更精确的实现。

参数:

xarray_like

输入数组。

axisint 或 int 元组，可选

计算值的轴。默认为 None，softmax 将在整个数组 x 上计算。

返回:

sndarray 或标量

一个与 x 形状相同的数组。结果的指数将在指定轴上求和为 1。如果 x 是标量，则返回标量。

注释

log_softmax 比输入使 softmax 饱和的 np.log(softmax(x)) 更准确（请参见下面的示例）。

在 1.5.0 版本中添加。

示例

在您的浏览器中尝试！

>>> import numpy as np
>>> from scipy.special import log_softmax
>>> from scipy.special import softmax
>>> np.set_printoptions(precision=5)

>>> x = np.array([1000.0, 1.0])

>>> y = log_softmax(x)
>>> y
array([   0., -999.])

>>> with np.errstate(divide='ignore'):
...   y = np.log(softmax(x))
...
>>> y
array([  0., -inf])

返回在新标签页中打开