jax.nn.celu#

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

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

计算逐元素函数

$$\begin{array}{r}\mathrm{celu}(x)=\{\begin{array}{ll}x,& x>0\\ \alpha (\exp (\frac{x}{\alpha })-1),& x\le 0\end{array}\end{array}$$

更多信息，请参阅 Continuously Differentiable Exponential Linear Units。

参数:

• x (ArrayLike) – 输入数组

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

返回值:

一个数组。

返回类型:

Array