mlx.nn.CELU

class CELU(alpha=1.0)

套用連續可微的指數線性單元（CELU）。

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

See celu() for the functional equivalent.

參數:

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

方法