mlx_rs::nn

Function silu

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

Applies the Sigmoid Linear Unit. Also known as Swish.

This is:

x * sigmoid(x)