mlx_rs::nn

Function celu

Source
pub fn celu(
    x: impl AsRef<Array>,
    alpha: impl Into<Option<f32>>,
) -> Result<Array>
Expand description

Applies the Continuously Differentiable Exponential Linear Unit.

This is:

maximum(x, 0) + alpha * (exp(minimum(x, 0) / alpha) - 1)