mlx.nn.CELU

class CELU(alpha=1.0)

应用连续可微指数线性单元 (CELU)。

逐元素应用 \(\max(0, x) + \min(0, \alpha * (\exp(x / \alpha) - 1))\)。

请参阅函数等效项 celu()。

参数:

alpha – CELU 公式中的 \(\alpha\) 值。默认值: 1.0

方法