mlx.nn.ELU

目錄

mlx.nn.ELU#

class ELU(alpha=1.0)#
套用指數線性單元(ELU)。

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

See elu() for the functional equivalent.

參數:

alpha -- the \(\alpha\) value for the ELU formulation. Default: 1.0

方法