mlx_rs::nn

Function leaky_relu

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

Applies the Leaky Rectified Linear Unit.

neg_slope is default to 0.01 if not provided.

This is:

maximum(neg_slope * x, x)