jax.nn.hard_tanh

内容

jax.nn.hard_tanh#

jax.nn.hard_tanh(x)[source]#

\(\mathrm{tanh}\) 激活函数。

计算逐元素函数

\[\begin{split}\mathrm{hard\_tanh}(x) = \begin{cases} -1, & x < -1\\ x, & -1 \le x \le 1\\ 1, & 1 < x \end{cases}\end{split}\]
参数:

x (ArrayLike) – 输入数组

返回值:

一个数组。

返回类型:

Array