jax.nn.sparse_sigmoid

jax.nn.sparse_sigmoid(x)

稀疏 sigmoid 激活函数。

计算函数

\[\begin{split}\mathrm{sparse\_sigmoid}(x) = \begin{cases} 0, & x \leq -1\\ \frac{1}{2}(x+1), & -1 < x < 1 \\ 1, & 1 \leq x \end{cases}\end{split}\]

这是 sigmoid 激活函数的孪生函数，它确保输入小于 -1 时输出为 0，输入大于 1 时输出为 1，输入在 -1 和 1 之间时输出为线性值。它是 sparse_plus 的导数。

更多信息，请参阅 Learning with Fenchel-Young Losses（第 6.2 节）。

参数:

x (ArrayLike) – 输入数组

返回值:

一个数组。

返回类型:

数组

另请参阅

sigmoid()