Skip to main content

elu

mlx_rs::nn

Function elu 

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

Applies the Exponential Linear Unit.

This is:

select(x.gt(0), x, alpha * (exp(x) - 1))

ยงParams

  • x: The input array
  • alpha: Default to 1.0 if not provided