mlx_rs::nn

Function selu

Source
pub fn selu(x: impl AsRef<Array>) -> Result<Array>
Expand description

Applies the Scaled Exponential Linear Unit.

This is:

elu(x, 1.67326) * 1.0507