jax.nn.稀疏_sigmoid

jax.nn.稀疏_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 的导数。

更多信息,请参见 使用 Fenchel-Young 损失学习 (第 6.2 节)

参数:

x (ArrayLike) – 输入数组

返回:

一个数组。

返回类型:

Array

参见

sigmoid()