mlx.nn.ELU

目录

mlx.nn.ELU#

class ELU(alpha=1.0)#
应用指数线性单元(Exponential Linear Unit)。

简单来说,就是 mx.where(x > 0, x, alpha * (mx.exp(x) - 1))

请参阅 elu() 函数式等价用法。

参数:

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

方法