jax.nn.celu

内容

jax.nn.celu#

jax.nn.celu(x, alpha=1.0)[source]#

连续可微指数线性单元激活。

计算逐元素函数

\[\begin{split}\mathrm{celu}(x) = \begin{cases} x, & x > 0\\ \alpha \left(\exp(\frac{x}{\alpha}) - 1\right), & x \le 0 \end{cases}\end{split}\]

更多信息,请参阅 连续可微指数线性单元

参数:
  • x (ArrayLike) – 输入数组

  • alpha (ArrayLike) – 数组或标量(默认值:1.0)

返回值:

一个数组。

返回类型:

Array